Remove remaining unittest.TestCase uses

timhoffm · timhoffm · commit ecb54239b712 · 2019-05-25T15:22:42.000+02:00
diff --git a/lib/matplotlib/tests/test_transforms.py b/lib/matplotlib/tests/test_transforms.py
@@ -1,5 +1,3 @@
-import unittest
-
 import numpy as np
 from numpy.testing import (assert_allclose, assert_almost_equal,
                            assert_array_equal, assert_array_almost_equal)
@@ -219,8 +217,8 @@ def transform_path_non_affine(self, path):
         return self.real_trans.transform_path(path)
 
 
-class BasicTransformTests(unittest.TestCase):
-    def setUp(self):
+class BasicTransformTests():
+    def setup_method(self):
 
         self.ta1 = mtransforms.Affine2D(shorthand_name='ta1').rotate(np.pi / 2)
         self.ta2 = mtransforms.Affine2D(shorthand_name='ta2').translate(10, 0)
@@ -352,8 +350,8 @@ def test_affine_simplification(self):
         assert_array_equal(expected_result, result)
 
 
-class TestTransformPlotInterface(unittest.TestCase):
-    def tearDown(self):
+class TestTransformPlotInterface:
+    def teardown_method(self):
         plt.close()
 
     def test_line_extent_axes_coords(self):
@@ -381,7 +379,6 @@ def test_line_extent_compound_coords1(self):
         assert_array_equal(ax.dataLim.get_points(),
                            np.array([[np.inf, -5.],
                                      [-np.inf, 35.]]))
-        plt.close()
 
     def test_line_extent_predata_transform_coords(self):
         # a simple line in (offset + data) coordinates
@@ -390,7 +387,6 @@ def test_line_extent_predata_transform_coords(self):
         ax.plot([0.1, 1.2, 0.8], [35, -5, 18], transform=trans)
         assert_array_equal(ax.dataLim.get_points(),
                            np.array([[1., -50.], [12., 350.]]))
-        plt.close()
 
     def test_line_extent_compound_coords2(self):
         # a simple line in (offset + data) coordinates in the y component, and
@@ -401,7 +397,6 @@ def test_line_extent_compound_coords2(self):
         ax.plot([0.1, 1.2, 0.8], [35, -5, 18], transform=trans)
         assert_array_equal(ax.dataLim.get_points(),
                            np.array([[np.inf, -50.], [-np.inf, 350.]]))
-        plt.close()
 
     def test_line_extents_affine(self):
         ax = plt.axes()
