Accept str for rules update on Cube object

TM1py/Objects/Cube.py

@@ -2,7 +2,7 @@
 
 import collections
 import json
-from typing import Iterable, List, Dict, Optional
+from typing import Iterable, List, Dict, Optional, Union
 
 from TM1py.Objects.Rules import Rules
 from TM1py.Objects.TM1Object import TM1Object
@@ -14,7 +14,7 @@ class Cube(TM1Object):
 
     """
 
-    def __init__(self, name: str, dimensions: Iterable[str], rules: Optional[Rules] = None):
+    def __init__(self, name: str, dimensions: Iterable[str], rules: Optional[Union[str, Rules]] = None):
         """
 
         :param name: name of the Cube
@@ -48,8 +48,13 @@ def rules(self) -> Rules:
         return self._rules
 
     @rules.setter
-    def rules(self, value: Rules):
-        self._rules = value
+    def rules(self, value: Union[str, Rules]):
+        if isinstance(value, str):
+            self._rules = Rules(rules=value)
+        elif isinstance(value, Rules):
+            self._rules = value
+        else:
+            raise ValueError('value must be of type str or Rules')
 
     @property
     def skipcheck(self) -> bool:

Tests/Cube_test.py

@@ -0,0 +1,33 @@
+import unittest
+
+from TM1py import Cube, Rules
+
+
+class TestCube(unittest.TestCase):
+    rules = """
+['d1':'e1'] = N: 1;
+['d1':'e2'] = N: 2;
+['d1':'e3'] = N: 3;
+"""
+    cube = Cube(
+        name="c1",
+        dimensions=["d1", "d2"],
+        rules=rules)
+
+    def test_update_rule_with_str(self):
+        self.cube.rules = "['d1':'e1'] = N: 1;"
+
+        self.assertEqual(
+            self.cube.rules,
+            Rules("['d1':'e1'] = N: 1;"))
+
+    def test_update_rule_with_rules_obj(self):
+        self.cube.rules = Rules("['d1':'e1'] = N: 2;")
+
+        self.assertEqual(
+            self.cube.rules,
+            Rules("['d1':'e1'] = N: 2;"))
+
+
+if __name__ == '__main__':
+    unittest.main()