Merge pull request #205 from namannimmo10/strcmp

Make runtime string comparison work

Author: certik

integration_tests/test_builtin_str_02.py

@@ -1,4 +1,4 @@
-def _lpython_str_equal(a: str, b: str) -> bool:
+def _lpython_strcmp_eq(a: str, b: str) -> bool:
     if len(a) != len(b):
         return False
     i: i32
@@ -7,13 +7,77 @@ def _lpython_str_equal(a: str, b: str) -> bool:
             return False
     return True
 
+def _lpython_strcmp_noteq(a: str, b: str) -> bool:
+    return not _lpython_strcmp_eq(a, b)
+
+def _lpython_strcmp_lt(a: str, b: str) -> bool:
+    l: i32
+    l = len(a)
+    if l > len(b):
+        l = len(b)
+    i: i32
+    for i in range(l):
+        if a[i] > b[i]:
+            return False
+    return True
+
+def _lpython_strcmp_lteq(a: str, b: str) -> bool:
+    return _lpython_strcmp_eq(a, b) or _lpython_strcmp_lt(a, b)
+
+def _lpython_strcmp_gt(a: str, b: str) -> bool:
+    l: i32
+    l = len(a)
+    if l > len(b):
+        l = len(b)
+    i: i32
+    for i in range(l):
+        if a[i] < b[i]:
+            return False
+    return True
+
+def _lpython_strcmp_gteq(a: str, b: str) -> bool:
+    return _lpython_strcmp_eq(a, b) or _lpython_strcmp_gt(a, b)
+
 def f():
-    assert _lpython_str_equal("a", "a")
-    assert not _lpython_str_equal("a2", "a")
-    assert not _lpython_str_equal("a", "a123")
-    assert not _lpython_str_equal("a23", "a24")
-    assert _lpython_str_equal("a24", "a24")
-    assert _lpython_str_equal("abcdefg", "abcdefg")
-    assert not _lpython_str_equal("abcdef3", "abcdefg")
+    assert _lpython_strcmp_eq("a", "a")
+    assert not _lpython_strcmp_eq("a2", "a")
+    assert not _lpython_strcmp_eq("a", "a123")
+    assert not _lpython_strcmp_eq("a23", "a24")
+    assert _lpython_strcmp_eq("a24", "a24")
+    assert _lpython_strcmp_eq("abcdefg", "abcdefg")
+    assert not _lpython_strcmp_eq("abcdef3", "abcdefg")
+
+    assert _lpython_strcmp_noteq("a2", "a")
+    assert _lpython_strcmp_noteq("a", "a123")
+    assert _lpython_strcmp_noteq("a23", "a24")
+    assert _lpython_strcmp_noteq("abcdef3", "abcdefg")
+    assert not _lpython_strcmp_noteq("a24", "a24")
+
+    assert _lpython_strcmp_lt("a", "a2")
+    assert _lpython_strcmp_lt("a", "a123")
+    assert _lpython_strcmp_lt("a23", "a24")
+    assert _lpython_strcmp_lt("abcdef3", "gbc")
+    assert not _lpython_strcmp_lt("bg", "abc")
+    assert not _lpython_strcmp_lt("abcdefg", "abcdef3")
+
+    assert _lpython_strcmp_lteq("a", "a2")
+    assert _lpython_strcmp_lteq("a123", "a123")
+    assert _lpython_strcmp_lteq("a23", "a24")
+    assert not _lpython_strcmp_lteq("abcdefg", "abcdef3")
+
+    assert _lpython_strcmp_gt("a2", "a")
+    assert _lpython_strcmp_gt("a123", "a")
+    assert _lpython_strcmp_gt("a24", "a23")
+    assert _lpython_strcmp_gt("z", "abcdef3")
+    assert not _lpython_strcmp_gt("abcdef3", "abcdefg")
+
+    a: str
+    a = "abcdefg"
+    b: str
+    b = "abcdef3"
+    assert _lpython_strcmp_gteq("a2", "a")
+    assert _lpython_strcmp_gteq("a123", "a123")
+    assert _lpython_strcmp_gteq("bg", "abc")
+    assert _lpython_strcmp_gteq(a, b)
 
 f()