Fix MemoryError in string operators (erp12#159)

TheOnlyEmily · Oct 14, 2021 · 30a49f2 · 30a49f2
1 parent abb6068
commit 30a49f2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pyshgp/push/types.py b/pyshgp/push/types.py
@@ -103,7 +103,7 @@ def __init__(self):
 class PushStrType(PushType):
 
     def __init__(self):
-        super().__init__("str", (str, np.str_, np.object_))
+        super().__init__("str", (str, np.str_, np.object_), is_collection=True)
 
 
 class PushVectorType(PushType):

diff --git a/tests/push/test_stack.py b/tests/push/test_stack.py
@@ -65,3 +65,8 @@ def test_set_nth_oob(self, str_stack: PushStack):
     def test_flush(self, int_stack: PushStack):
         int_stack.push(1).push(-1).flush()
         assert len(int_stack) == 0
+
+    def test_large_str(self, str_stack: PushStack):
+        s = "largestr"*1000
+        str_stack.push(s)
+        assert len(str_stack.pop()) != len(s)