Fix PEP8 Issues

poliastro · Dec 6, 2018 · 8598713 · 8598713
1 parent 43804a9
commit 8598713
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/poliastro/tests/test_plotting2d.py b/src/poliastro/tests/test_plotting2d.py
@@ -9,10 +9,12 @@
 from poliastro.bodies import Earth, Mars, Sun
 from poliastro.twobody.orbit import Orbit
 
+
 @pytest.fixture(scope="module")
 def op():
     return OrbitPlotter2D()
 
+
 def test_get_figure_has_expected_properties(op):
     figure = op.figure
 
@@ -85,7 +87,7 @@ def test_savefig_calls_prepare_plot(op, mock_prepare_plot, mock_export):
     earth = Orbit.from_body_ephem(Earth)
     op.plot(orbit=earth, label="Obj")
     with tempfile.NamedTemporaryFile() as fp:
-        m.savefig(filename=fp.name + ".jpeg")
+        op.savefig(filename=fp.name + ".jpeg")
 
     assert mock_export.call_count == 1
     mock_prepare_plot.assert_called_once_with()