objset: Add frozenset tests, skippable if frozenset not available.

1 files changed, 36 insertions, 0 deletions

diff --git a/tests/basics/frozenset_binop.py b/tests/basics/frozenset_binop.py
new file mode 100644
index 000000000..5cc07e9e1
--- /dev/null
+++ b/tests/basics/frozenset_binop.py
@@ -0,0 +1,36 @@
+try:
+    frozenset
+except NameError:
+    print("SKIP")
+    import sys
+    sys.exit()
+
+sets = [
+    frozenset(), frozenset({1}), frozenset({1, 2}), frozenset({1, 2, 3}), frozenset({2, 3}),
+    frozenset({2, 3, 5}), frozenset({5}), frozenset({7})
+]
+for s in sets:
+    for t in sets:
+        print(sorted(s), '|', sorted(t), '=', sorted(s | t))
+        print(sorted(s), '^', sorted(t), '=', sorted(s ^ t))
+        print(sorted(s), '&', sorted(t), '=', sorted(s & t))
+        print(sorted(s), '-', sorted(t), '=', sorted(s - t))
+        u = s.copy()
+        u |= t
+        print(sorted(s), "|=", sorted(t), '-->', sorted(u))
+        u = s.copy()
+        u ^= t
+        print(sorted(s), "^=", sorted(t), '-->', sorted(u))
+        u = s.copy()
+        u &= t
+        print(sorted(s), "&=", sorted(t), "-->", sorted(u))
+        u = s.copy()
+        u -= t
+        print(sorted(s), "-=", sorted(t), "-->", sorted(u))
+
+        print(sorted(s), '==', sorted(t), '=', s == t)
+        print(sorted(s), '!=', sorted(t), '=', s != t)
+        print(sorted(s), '>', sorted(t), '=', s > t)
+        print(sorted(s), '>=', sorted(t), '=', s >= t)
+        print(sorted(s), '<', sorted(t), '=', s < t)
+        print(sorted(s), '<=', sorted(t), '=', s <= t)