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

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/basics/frozenset_difference.py b/tests/basics/frozenset_difference.py
new file mode 100644
index 0000000000..3d142f9595
--- /dev/null
+++ b/tests/basics/frozenset_difference.py
@@ -0,0 +1,21 @@
+try:
+    frozenset
+except NameError:
+    print("SKIP")
+    import sys
+    sys.exit()
+
+l = [1, 2, 3, 4]
+s = frozenset(l)
+outs = [s.difference(),
+        s.difference(frozenset({1})),
+        s.difference(frozenset({1}), [1, 2]),
+        s.difference(frozenset({1}), {1, 2}, {2, 3})]
+for out in outs:
+    print(type(out), sorted(out))
+
+s = frozenset(l)
+try:
+    print(s.difference_update({1}))
+except AttributeError:
+    print("AttributeError")