From 0834721390ec539406e2cb2830aabda34614d10e Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 20 Jan 2021 08:54:53 +0000 Subject: [PATCH] remove use of pytest in test code --- .../functional/functional_cpu_test.py | 11 +++++++---- .../torchaudio_unittest/librosa_compatibility_test.py | 2 -- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/test/torchaudio_unittest/functional/functional_cpu_test.py b/test/torchaudio_unittest/functional/functional_cpu_test.py index 4073654011..c4ba42391d 100644 --- a/test/torchaudio_unittest/functional/functional_cpu_test.py +++ b/test/torchaudio_unittest/functional/functional_cpu_test.py @@ -1,10 +1,10 @@ import math +import warnings import torch import torchaudio import torchaudio.functional as F from parameterized import parameterized -import pytest import itertools from torchaudio_unittest import common_utils @@ -33,17 +33,20 @@ class TestSpectrogramFloat64(Spectrogram, common_utils.PytorchTestCase): class TestCreateFBMatrix(common_utils.TorchaudioTestCase): def test_no_warning_high_n_freq(self): - with pytest.warns(None) as w: + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter("always") F.create_fb_matrix(288, 0, 8000, 128, 16000) assert len(w) == 0 def test_no_warning_low_n_mels(self): - with pytest.warns(None) as w: + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter("always") F.create_fb_matrix(201, 0, 8000, 89, 16000) assert len(w) == 0 def test_warning(self): - with pytest.warns(None) as w: + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter("always") F.create_fb_matrix(201, 0, 8000, 128, 16000) assert len(w) == 1 diff --git a/test/torchaudio_unittest/librosa_compatibility_test.py b/test/torchaudio_unittest/librosa_compatibility_test.py index e1ef80fa79..92d5baff06 100644 --- a/test/torchaudio_unittest/librosa_compatibility_test.py +++ b/test/torchaudio_unittest/librosa_compatibility_test.py @@ -17,8 +17,6 @@ import librosa import scipy -import pytest - from torchaudio_unittest import common_utils