test: fix name test

wcxve · Mar 1, 2024 · edf0015 · edf0015
1 parent 6880fa6
commit edf0015
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/tests/model/test_name.py b/tests/model/test_name.py
@@ -1,6 +1,6 @@
-from elisa.model.add import Powerlaw
+from elisa.model.add import PowerLaw
 
 
 def test_model_name():
-    model = Powerlaw() + Powerlaw()
-    assert repr(model) == "powerlaw + powerlaw2"
+    model = PowerLaw() + PowerLaw()
+    assert model.name == 'powerlaw + powerlaw_2'
diff --git a/tests/parameter/test_name.py b/tests/parameter/test_name.py
@@ -1,9 +1,9 @@
-from elisa.model.base import UniformParameter
+from elisa.model.parameter import UniformParameter
 
 
 def test_param_name():
-    a = UniformParameter("a", "a", 1.0, 0.0, 2.0)
-    a2 = UniformParameter("a2", "a2", 1.0, 0.0, 2.0)
+    a = UniformParameter('a', 1.5, 0.0, 2.0)
+    a2 = UniformParameter('a', 0.5, 0.0, 2.0)
     b = a + a2
-    assert repr(b) == "a + a2"
+    assert str(b) == "a + a'"
     assert b.default == 2.0