init

.venv/lib/python3.8/site-packages/mypyc/test-data/run-sets.test

@@ -0,0 +1,117 @@
+# Test cases for sets (compile and run)
+
+[case testSets]
+from typing import Set, List
+def instantiateLiteral() -> Set[int]:
+    return {1, 2, 3, 5, 8}
+
+def fromIterator() -> List[Set[int]]:
+    a = set([1, 3, 5])
+    b = set((1, 3, 5))
+    c = set({1: '1', 3: '3', 5: '5'})
+    d = set(x for x in range(1, 6, 2))
+    e = set((x for x in range(1, 6, 2)))
+    return [a, b, c, d, e]
+
+def fromIterator2() -> Set[int]:
+    tmp_list = [1, 2, 3, 4, 5]
+    return set((x + 1) for x in ((y * 10) for y in (z for z in tmp_list if z < 4)))
+
+def addIncrementing(s : Set[int]) -> None:
+    for a in [1, 2, 3]:
+        if a not in s:
+            s.add(a)
+            return
+
+def replaceWith1(s : Set[int]) -> None:
+    s.clear()
+    s.add(1)
+
+def remove1(s : Set[int]) -> None:
+    s.remove(1)
+
+def discard1(s: Set[int]) -> None:
+    s.discard(1)
+
+def pop(s : Set[int]) -> int:
+    return s.pop()
+
+def update(s: Set[int], x: List[int]) -> None:
+    s.update(x)
+
+[file driver.py]
+from native import instantiateLiteral
+from testutil import assertRaises
+
+val = instantiateLiteral()
+assert 1 in val
+assert 2 in val
+assert 3 in val
+assert 5 in val
+assert 8 in val
+assert len(val) == 5
+assert val == {1, 2, 3, 5, 8}
+s = 0
+for i in val:
+    s += i
+assert s == 19
+
+from native import fromIterator
+sets = fromIterator()
+for s in sets:
+    assert s == {1, 3, 5}
+
+from native import fromIterator2
+s = fromIterator2()
+assert s == {11, 21, 31}
+
+from native import addIncrementing
+s = set()
+addIncrementing(s)
+assert s == {1}
+addIncrementing(s)
+assert s == {1, 2}
+addIncrementing(s)
+assert s == {1, 2, 3}
+
+from native import replaceWith1
+s = {3, 7, 12}
+replaceWith1(s)
+assert s == {1}
+
+from native import remove1
+import traceback
+s = {1, 4, 6}
+remove1(s)
+assert s == {4, 6}
+with assertRaises(KeyError, '1'):
+    remove1(s)
+
+from native import discard1
+s = {1, 4, 6}
+discard1(s)
+assert s == {4, 6}
+discard1(s)
+assert s == {4, 6}
+
+from native import pop
+s = {1, 2, 3}
+x = pop(s)
+assert len(s) == 2
+assert x in [1, 2, 3]
+y = pop(s)
+assert len(s) == 1
+assert y in [1, 2, 3]
+assert x != y
+z = pop(s)
+assert len(s) == 0
+assert z in [1, 2, 3]
+assert x != z
+assert y != z
+with assertRaises(KeyError, 'pop from an empty set'):
+    pop(s)
+
+from native import update
+s = {1, 2, 3}
+update(s, [5, 4, 3])
+assert s == {1, 2, 3, 4, 5}