From b899d92a30684acdbceec8cb4bcfdc5bdd9f8c44 Mon Sep 17 00:00:00 2001
From: HugoMVale <57530119+HugoMVale@users.noreply.github.com>
Date: Thu, 2 Nov 2023 22:57:52 +0100
Subject: [PATCH] extend plot test

---
 tests/copolymerization/test_copolymerization.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/copolymerization/test_copolymerization.py b/tests/copolymerization/test_copolymerization.py
index bd33393..10d2225 100644
--- 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