FIX: Serialize unary op

fumitoh · Oct 5, 2024 · f859ffb · f859ffb
1 parent 1dd92c7
commit f859ffb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/modelx/serialize/serializer_6.py b/modelx/serialize/serializer_6.py
@@ -1493,7 +1493,10 @@ def condition(cls, node):
         return True
 
     def decode(self):
-        valstr = self.node.first_token.string.strip()
+        if isinstance(self.node, ast.UnaryOp):  # such as -1
+            valstr = self.node.first_token.string.strip() + self.node.last_token.string.strip()
+        else:
+            valstr = self.node.first_token.string.strip()
         if valstr in ["True", "False", "None"]:
             return ast.literal_eval(self.node)
         else:

diff --git a/modelx/tests/serialize/test_primitive_types.py b/modelx/tests/serialize/test_primitive_types.py
@@ -9,7 +9,8 @@
     ('b', False, 'False'),
     ('c', None, 'None'),
     ('d', np.inf, 'Infinity'),
-    ('e', np.pi, '3.141592653589793')
+    ('e', np.pi, '3.141592653589793'),
+    ('f', -1, '-1')
 ]