extend plot test

HugoMVale · Nov 2, 2023 · b899d92 · b899d92
1 parent 1275095
commit b899d92
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/copolymerization/test_copolymerization.py b/tests/copolymerization/test_copolymerization.py
@@ -121,6 +121,8 @@ def test_TerminalCopoModel_drift():
 
 
 def test_TerminalCopoModel_plot():
-    model = TerminalCopoModel(0.5, 0.5, "Monomer1", "Monomer2", "Model1")
-    result = model.plot(M=1, f0=0.2, title="Test Plot", return_objects=True)
+    model = TerminalCopoModel(0.5, 0.5, "Monomer1", "Monomer2", "MyModel")
+    result = model.plot(M=1, f0=0.2, return_objects=True)
+    assert len(result) == 2
+    result = model.plot(M=1, return_objects=True)
     assert len(result) == 2