fix pyplot test

jhavl · jhavl · commit 7951daeba9a0 · 2021-09-13T12:46:33.000+10:00
diff --git a/tests/test_PyPlot.py b/tests/test_PyPlot.py
@@ -6,15 +6,16 @@
 # import numpy.testing as nt
 # import numpy as np
 import roboticstoolbox as rp
+
 # import spatialmath as sm
 import unittest
 
 
 class TestPyPlot(unittest.TestCase):
-
     def test_PyPlot(self):
         panda = rp.models.DH.Panda()
         from roboticstoolbox.backends.PyPlot import PyPlot
+
         env = PyPlot()
         env.launch()
         env.add(panda)
@@ -25,6 +26,7 @@ def test_PyPlot(self):
     def test_PyPlot_invisible(self):
         panda = rp.models.DH.Panda()
         from roboticstoolbox.backends.PyPlot import PyPlot
+
         env = PyPlot()
         env.launch()
         env.add(panda, display=False)
@@ -35,13 +37,14 @@ def test_PyPlot_invisible(self):
     def test_unimplemented(self):
         # TODO remove these as implemented
         from roboticstoolbox.backends.PyPlot import PyPlot
+
         env = PyPlot()
         env.reset()
 
         env.restart()
-        env.remove()
+        env.remove(0)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
 
     unittest.main()
