Browse Source

Add test for gcodeListDecorator

It's super simple, but hey, it's also easy to write.
Jaime van Kessel 6 years ago
parent
commit
67a63e7954
2 changed files with 18 additions and 2 deletions
  1. 2 2
      cura/Scene/GCodeListDecorator.py
  2. 16 0
      tests/TestGCodeListDecorator.py

+ 2 - 2
cura/Scene/GCodeListDecorator.py

@@ -10,10 +10,10 @@ class GCodeListDecorator(SceneNodeDecorator):
     def getGCodeList(self) -> List[str]:
         return self._gcode_list
 
-    def setGCodeList(self, list: List[str]):
+    def setGCodeList(self, list: List[str]) -> None:
         self._gcode_list = list
 
     def __deepcopy__(self, memo) -> "GCodeListDecorator":
         copied_decorator = GCodeListDecorator()
         copied_decorator.setGCodeList(self.getGCodeList())
-        return copied_decorator
+        return copied_decorator

+ 16 - 0
tests/TestGCodeListDecorator.py

@@ -0,0 +1,16 @@
+from cura.Scene.GCodeListDecorator import GCodeListDecorator
+
+
+def test_setAndGetList():
+    decorator = GCodeListDecorator()
+
+    decorator.setGCodeList(["Test"])
+    assert decorator.getGCodeList() == ["Test"]
+
+
+def test_copyGCodeDecorator():
+    decorator = GCodeListDecorator()
+    decorator.setGCodeList(["Test"])
+    import copy
+    copied_decorator = copy.deepcopy(decorator)
+    assert decorator.getGCodeList() == copied_decorator.getGCodeList()