Merge pull request #1451 from Vyxal/fix-untruth

Make untruth work on `0` and `[]`
Vyxal · Oct 2, 2022 · 02127e3 · 02127e3
2 parents 333834a + e512a7b
commit 02127e3
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 1 deletion.
diff --git a/documents/knowledge/elements.yaml b/documents/knowledge/elements.yaml
@@ -5437,6 +5437,7 @@
       - "[[0,3,4,6]] : [1,0,0,1,1,0,1]"
       - "[[[0,1,0],[2,0,3],[1,0,1]]] : [[[0, 0, 0, 0], [1, 0, 0, 0]], [[0, 1, 0, 0], [0, 0, 0, 0]], [[0, 0, 0, 1], [0, 0, 0, 0]]]"
       - "[[]] : []"
+      - "[0] : [1]"
 
 - element: "ÞǓ"
   name: Connected Uniquify

diff --git a/tests/test_elements.py b/tests/test_elements.py
diff --git a/vyxal/elements.py b/vyxal/elements.py
@@ -5704,7 +5704,9 @@ def untruth(lhs, ctx):
                 ref = ref[x[i]]
             ref[x[dimensions - 1]] = 1
         return matrix
-    return [int(x in lhs) for x in range((monadic_maximum(lhs, ctx) or -1) + 1)]
+    if not lhs and lhs != 0:
+        return []
+    return [int(x in lhs) for x in range(monadic_maximum(lhs, ctx) + 1)]
 
 
 def unwrap(lhs, ctx):