diff options
author | Charles Harris <charlesr.harris@gmail.com> | 2015-07-05 10:18:57 -0600 |
---|---|---|
committer | Charles Harris <charlesr.harris@gmail.com> | 2015-07-05 10:35:01 -0600 |
commit | ad40c230c4da2ca336bed6b093e8efaba590eec3 (patch) | |
tree | 18abebb0ba7b594a122a690c1866be75cae2fdea /numpy/testing/tests/test_utils.py | |
parent | c80c609950fe5aeecf3082d397c7b5149a834fca (diff) | |
download | numpy-ad40c230c4da2ca336bed6b093e8efaba590eec3.tar.gz |
MAINT: pyflakes for numpy/testing, numpy/testing/tests.
Diffstat (limited to 'numpy/testing/tests/test_utils.py')
-rw-r--r-- | numpy/testing/tests/test_utils.py | 32 |
1 files changed, 27 insertions, 5 deletions
diff --git a/numpy/testing/tests/test_utils.py b/numpy/testing/tests/test_utils.py index e75914833..a31fce4af 100644 --- a/numpy/testing/tests/test_utils.py +++ b/numpy/testing/tests/test_utils.py @@ -9,17 +9,19 @@ from numpy.testing import ( assert_array_almost_equal, build_err_msg, raises, assert_raises, assert_warns, assert_no_warnings, assert_allclose, assert_approx_equal, assert_array_almost_equal_nulp, assert_array_max_ulp, - clear_and_catch_warnings, run_module_suite) + clear_and_catch_warnings, run_module_suite + ) import unittest + class _GenericTest(object): + def _test_equal(self, a, b): self._assert_func(a, b) def _test_not_equal(self, a, b): try: self._assert_func(a, b) - passed = True except AssertionError: pass else: @@ -61,7 +63,9 @@ class _GenericTest(object): def test_array_likes(self): self._test_equal([1, 2, 3], (1, 2, 3)) + class TestArrayEqual(_GenericTest, unittest.TestCase): + def setUp(self): self._assert_func = assert_array_equal @@ -139,7 +143,9 @@ class TestArrayEqual(_GenericTest, unittest.TestCase): self._test_not_equal(c, b) + class TestBuildErrorMessage(unittest.TestCase): + def test_build_err_msg_defaults(self): x = np.array([1.00001, 2.00002, 3.00003]) y = np.array([1.00002, 2.00003, 3.00004]) @@ -182,7 +188,9 @@ class TestBuildErrorMessage(unittest.TestCase): '1.000000002, 2.00003 , 3.00004 ])') self.assertEqual(a, b) + class TestEqual(TestArrayEqual): + def setUp(self): self._assert_func = assert_equal @@ -217,7 +225,9 @@ class TestEqual(TestArrayEqual): self._assert_func(x, x) self._test_not_equal(x, y) + class TestArrayAlmostEqual(_GenericTest, unittest.TestCase): + def setUp(self): self._assert_func = assert_array_almost_equal @@ -259,6 +269,7 @@ class TestArrayAlmostEqual(_GenericTest, unittest.TestCase): class TestAlmostEqual(_GenericTest, unittest.TestCase): + def setUp(self): self._assert_func = assert_almost_equal @@ -321,7 +332,9 @@ class TestAlmostEqual(_GenericTest, unittest.TestCase): # remove anything that's not the array string self.assertEqual(str(e).split('%)\n ')[1], b) + class TestApproxEqual(unittest.TestCase): + def setUp(self): self._assert_func = assert_approx_equal @@ -368,7 +381,9 @@ class TestApproxEqual(unittest.TestCase): self.assertRaises(AssertionError, lambda: self._assert_func(ainf, anan)) + class TestRaises(unittest.TestCase): + def setUp(self): class MyException(Exception): pass @@ -382,11 +397,11 @@ class TestRaises(unittest.TestCase): pass def test_correct_catch(self): - f = raises(self.e)(self.raises_exception)(self.e) + raises(self.e)(self.raises_exception)(self.e) # raises? def test_wrong_exception(self): try: - f = raises(self.e)(self.raises_exception)(RuntimeError) + raises(self.e)(self.raises_exception)(RuntimeError) # raises? except RuntimeError: return else: @@ -394,13 +409,15 @@ class TestRaises(unittest.TestCase): def test_catch_no_raise(self): try: - f = raises(self.e)(self.does_not_raise_exception)() + raises(self.e)(self.does_not_raise_exception)() # raises? except AssertionError: return else: raise AssertionError("should have raised an AssertionError") + class TestWarns(unittest.TestCase): + def test_warn(self): def f(): warnings.warn("yo") @@ -436,7 +453,9 @@ class TestWarns(unittest.TestCase): if failed: raise AssertionError("wrong warning caught by assert_warn") + class TestAssertAllclose(unittest.TestCase): + def test_simple(self): x = 1e-3 y = 1e-9 @@ -473,6 +492,7 @@ class TestAssertAllclose(unittest.TestCase): msg = exc.args[0] self.assertTrue("mismatch 25.0%" in msg) + class TestArrayAlmostEqualNulp(unittest.TestCase): def test_float64_pass(self): @@ -641,6 +661,7 @@ class TestArrayAlmostEqualNulp(unittest.TestCase): class TestULP(unittest.TestCase): + def test_equal(self): x = np.random.randn(10) assert_array_max_ulp(x, x, maxulp=0) @@ -743,6 +764,7 @@ def test_clear_and_catch_warnings(): class my_cacw(clear_and_catch_warnings): + class_modules = (sys.modules[__name__],) |