add teardown method to close any open windows

Author: petercorke

spatialmath/base/test_transforms.py

@@ -52,6 +52,10 @@
 
 class TestVector(unittest.TestCase):
 
+    @classmethod
+    def tearDownClass(cls):
+        plt.close('all')
+
     def test_unit(self):
 
         nt.assert_array_almost_equal(unitvec([1, 0, 0]), np.r_[1, 0, 0])

spatialmath/test_pose2d.py

@@ -26,6 +26,9 @@ def array_compare(x, y):
 
 class TestSO2(unittest.TestCase):
 
+    @classmethod
+    def tearDownClass(cls):
+        plt.close('all')
 
     def test_constructor(self):
 
@@ -224,7 +227,11 @@ def test_plot(self):
 
 
 class TestSE2(unittest.TestCase):
- 
+
+    @classmethod
+    def tearDownClass(cls):
+        plt.close('all')
+
     def test_constructor(self):
 
         self.assertIsInstance(SE2(), SE2)

spatialmath/test_pose3d.py

@@ -25,6 +25,9 @@ def array_compare(x, y):
 
 
 class TestSO3(unittest.TestCase):
+    @classmethod
+    def tearDownClass(cls):
+        plt.close('all')
 
     def test_constructor(self):
 
@@ -569,6 +572,10 @@ def test_functions_vect(self):
 
 class TestSE3(unittest.TestCase):
 
+    @classmethod
+    def tearDownClass(cls):
+        plt.close('all')
+
     def test_constructor(self):
 
         # null constructor