def test_init(self):
import _graphillion
self.assertEqual(_graphillion._elem_limit(), 2**20 - 1)
setset.set_universe([])
self.assertEqual(setset._obj2int, {})
self.assertEqual(setset._int2obj, [None])
self.assertEqual(setset.universe(), [])
setset.set_universe(['i', 'ii'])
self.assertEqual(setset._obj2int, {'i': 1, 'ii': 2})
self.assertEqual(setset._int2obj, [None, 'i', 'ii'])
self.assertEqual(setset.universe(), ['i', 'ii'])
ss = setset({})
self.assertEqual(
ss, setset([set(),
set(['i']),
set(['i', 'ii']),
set(['ii'])]))
setset.set_universe(['1'])
self.assertEqual(setset._obj2int, {'1': 1})
self.assertEqual(setset._int2obj, [None, '1'])
self.assertEqual(setset.universe(), ['1'])
ss = setset({})
self.assertEqual(ss, setset([s0, s1]))
self.assertRaises(ValueError, setset.set_universe, ['1', '1'])
def test_init(self):
setset.set_universe([])
self.assertEqual(setset._obj2int, {})
self.assertEqual(setset._int2obj, [None])
self.assertEqual(setset.universe(), [])
setset.set_universe(['i', 'ii'])
self.assertEqual(setset._obj2int, {'i': 1, 'ii': 2})
self.assertEqual(setset._int2obj, [None, 'i', 'ii'])
self.assertEqual(setset.universe(), ['i', 'ii'])
ss = setset({})
self.assertEqual(
ss, setset([set(),
set(['i']),
set(['i', 'ii']),
set(['ii'])]))
setset.set_universe(['1'])
self.assertEqual(setset._obj2int, {'1': 1})
self.assertEqual(setset._int2obj, [None, '1'])
self.assertEqual(setset.universe(), ['1'])
ss = setset({})
self.assertEqual(ss, setset([s0, s1]))
self.assertRaises(ValueError, setset.set_universe, ['1', '1'])
def test_init(self):
import _graphillion
self.assertEqual(_graphillion._elem_limit(), 2**20 - 1)
setset.set_universe([])
self.assertEqual(setset._obj2int, {})
self.assertEqual(setset._int2obj, [None])
self.assertEqual(setset.universe(), [])
setset.set_universe(['i', 'ii'])
self.assertEqual(setset._obj2int, {'i': 1, 'ii': 2})
self.assertEqual(setset._int2obj, [None, 'i', 'ii'])
self.assertEqual(setset.universe(), ['i', 'ii'])
ss = setset({})
self.assertEqual(
ss,
setset([set(), set(['i']), set(['i','ii']), set(['ii'])]))
setset.set_universe(['1'])
self.assertEqual(setset._obj2int, {'1': 1})
self.assertEqual(setset._int2obj, [None, '1'])
self.assertEqual(setset.universe(), ['1'])
ss = setset({})
self.assertEqual(ss, setset([s0, s1]))
self.assertRaises(ValueError, setset.set_universe, ['1', '1'])
def test_init(self):
setset.set_universe([])
self.assertEqual(setset._obj2int, {})
self.assertEqual(setset._int2obj, [None])
self.assertEqual(setset.universe(), [])
setset.set_universe(['i', 'ii'])
self.assertEqual(setset._obj2int, {'i': 1, 'ii': 2})
self.assertEqual(setset._int2obj, [None, 'i', 'ii'])
self.assertEqual(setset.universe(), ['i', 'ii'])
ss = setset({})
self.assertEqual(
ss,
setset([set(), set(['i']), set(['i','ii']), set(['ii'])]))
setset.set_universe(['1'])
self.assertEqual(setset._obj2int, {'1': 1})
self.assertEqual(setset._int2obj, [None, '1'])
self.assertEqual(setset.universe(), ['1'])
ss = setset({})
self.assertEqual(ss, setset([s0, s1]))
self.assertRaises(ValueError, setset.set_universe, ['1', '1'])
def capacity(self):
ss = setset()
self.assertFalse(ss)
ss = setset([s0, s12, s13])
self.assertTrue(ss)
self.assertEqual(len(ss), 3)
self.assertEqual(ss.len(), 3)
def capacity(self):
ss = setset()
self.assertFalse(ss)
ss = setset([s0, s12, s13])
self.assertTrue(ss)
self.assertEqual(len(ss), 3)
self.assertEqual(ss.len(), 3)
def test_iterators(self):
ss = setset([s0, s12, s13])
v = []
for s in ss:
v.append(s)
self.assertEqual(len(v), 3)
self.assertEqual(ss, setset(v))
v = []
for s in ss:
v.append(s)
self.assertEqual(len(v), 3)
self.assertEqual(ss, setset(v))
ss = setset([s1, s12, s13])
v = []
for s in ss:
v.append(s)
self.assertEqual(len(v), 3)
self.assertEqual(ss, setset(v))
ss1 = setset([s0, s12, s13])
ss2 = setset()
for s in ss1.rand_iter():
ss2 = ss2 | setset([s])
self.assertEqual(ss1, ss2)
gen = ss1.rand_iter()
self.assertTrue(isinstance(next(gen), set))
ss = setset([s0, s1, s12, s123, s1234, s134, s14, s4])
r = []
for s in ss.max_iter({'1': .3, '2': -.2, '3': -.2}, default=.4):
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], s14)
self.assertEqual(r[1], s134)
self.assertEqual(r[2], s4)
r = []
for s in ss.max_iter():
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], s1234)
self.assertEqual(r[-1], s0)
r = []
for s in ss.min_iter({'1': .3, '2': -.2, '3': -.2}, default=.4):
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], s123)
self.assertEqual(r[1], s0)
self.assertEqual(r[2], s12)
r = []
for s in ss.min_iter():
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], set())
self.assertEqual(r[-1], s1234)
def test_iterators(self):
ss = setset([s0, s12, s13])
v = []
for s in ss:
v.append(s)
self.assertEqual(len(v), 3)
self.assertEqual(ss, setset(v))
v = []
for s in ss:
v.append(s)
self.assertEqual(len(v), 3)
self.assertEqual(ss, setset(v))
ss = setset([s1, s12, s13])
v = []
for s in ss:
v.append(s)
self.assertEqual(len(v), 3)
self.assertEqual(ss, setset(v))
ss1 = setset([s0, s12, s13])
ss2 = setset()
for s in ss1.rand_iter():
ss2 = ss2 | setset([s])
self.assertEqual(ss1, ss2)
gen = ss1.rand_iter()
self.assertTrue(isinstance(gen.next(), set))
ss = setset([s0, s1, s12, s123, s1234, s134, s14, s4])
r = []
for s in ss.max_iter({'1': .3, '2': -.2, '3': -.2}, default=.4):
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], s14)
self.assertEqual(r[1], s134)
self.assertEqual(r[2], s4)
r = []
for s in ss.max_iter():
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], s1234)
self.assertEqual(r[-1], s0)
r = []
for s in ss.min_iter({'1': .3, '2': -.2, '3': -.2}, default=.4):
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], s123)
self.assertEqual(r[1], s0)
self.assertEqual(r[2], s12)
r = []
for s in ss.min_iter():
r.append(s)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], set())
self.assertEqual(r[-1], s1234)
def test_large(self):
n = 1000
setset.set_universe(range(n))
ss = setset({}) - setset([set([1]), set([1, 2])])
self.assertTrue(ss)
self.assertEqual(ss.len(), 10715086071862673209484250490600018105614048117055336074437503883703510511249361224931983788156958581275946729175531468251871452856923140435984577574698574803934567774824230985421074605062371141877954182153046474983581941267398767559165543946077062914571196477686542167660429831652624386837205668069374)
i = 0
for s in ss:
if i > 100: break
i += 1
def test_large(self):
n = 1000
setset.set_universe(range(n))
ss = setset({}) - setset([set([1]), set([1, 2])])
self.assertTrue(ss)
self.assertEqual(
ss.len(),
10715086071862673209484250490600018105614048117055336074437503883703510511249361224931983788156958581275946729175531468251871452856923140435984577574698574803934567774824230985421074605062371141877954182153046474983581941267398767559165543946077062914571196477686542167660429831652624386837205668069374
)
i = 0
for s in ss:
if i > 100: break
i += 1
def test_io(self):
ss = setset()
st = ss.dumps()
self.assertEqual(st, "B\n.\n")
ss = setset.loads(st)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset())
ss = setset([s0])
st = ss.dumps()
self.assertEqual(st, "T\n.\n")
ss = setset.loads(st)
self.assertEqual(ss, setset([s0]))
v = [s0, s1, s12, s123, s1234, s134, s14, s4]
ss = setset(v)
st = ss.dumps()
ss = setset.loads(st)
self.assertEqual(ss, setset(v))
# skip this test, becasue string is treated as an element
# ss = setset(st)
# self.assertEqual(ss, setset(v))
with tempfile.TemporaryFile() as f:
ss.dump(f)
f.seek(0)
ss = setset.load(f)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset(v))
def test_io(self):
ss = setset()
st = ss.dumps()
self.assertEqual(st, "B\n.\n")
ss = setset.loads(st)
self.assertEqual(ss, setset())
ss = setset([s0])
st = ss.dumps()
self.assertEqual(st, "T\n.\n")
ss = setset.loads(st)
self.assertEqual(ss, setset([s0]))
v = [s0, s1, s12, s123, s1234, s134, s14, s4]
ss = setset(v)
st = ss.dumps()
ss = setset.loads(st)
self.assertEqual(ss, setset(v))
# skip this test, becasue string is treated as an element
# ss = setset(st)
# self.assertEqual(ss, setset(v))
with tempfile.TemporaryFile() as f:
ss.dump(f)
f.seek(0)
ss = setset.load(f)
self.assertEqual(ss, setset(v))
def test_constructors(self):
ss = setset()
self.assertTrue(isinstance(ss, setset))
self.assertEqual(repr(ss), 'setset([])')
ss = setset([s0, s12, s13])
self.assertEqual(repr(ss),
"setset([set([]), set(['1', '2']), set(['1', '3'])])")
ss = setset([frozenset(s0), frozenset(s12), frozenset(s13)])
self.assertEqual(repr(ss),
"setset([set([]), set(['1', '2']), set(['1', '3'])])")
ss = setset({'include': list(s12), 'exclude': list(s4)})
self.assertEqual(repr(ss),
"setset([set(['1', '2']), set(['1', '2', '3'])])")
self.assertRaises(TypeError, setset, {'x': list(s1)})
self.assertRaises(TypeError, setset,
{'include': list(s12), 'exclude': list(s23)})
# copy constructor
ss1 = setset([s0, s12, s13])
ss2 = ss1.copy()
self.assertTrue(isinstance(ss2, setset))
ss1.clear()
self.assertEqual(repr(ss1), 'setset([])')
self.assertEqual(repr(ss2),
"setset([set([]), set(['1', '2']), set(['1', '3'])])")
# repr
ss = setset({})
self.assertEqual(
repr(ss),
"setset([set([]), set(['1']), set(['2']), set(['3']), set(['4']), set(['1', ' ...")
def test_probability(self):
p = {'1': .9, '2': .8, '3': .7, '4': .6}
ss = setset()
self.assertEqual(ss.probability(p), 0)
ss = setset([s0])
self.assertAlmostEqual(ss.probability(p), .0024)
ss = setset([s1])
self.assertAlmostEqual(ss.probability(p), .0216)
ss = setset([s2])
self.assertAlmostEqual(ss.probability(p), .0096)
ss = setset([s12, s13])
self.assertAlmostEqual(ss.probability(p), .1368)
ss = setset([s1234])
self.assertAlmostEqual(ss.probability(p), .3024)
ss = setset([s0, s1, s2, s12, s13, s1234])
self.assertAlmostEqual(ss.probability(p), .4728)
def test_probability(self):
p = {'1': .9, '2': .8, '3': .7, '4': .6}
ss = setset()
self.assertEqual(ss.probability(p), 0)
ss = setset([s0])
self.assertAlmostEqual(ss.probability(p), .0024)
ss = setset([s1])
self.assertAlmostEqual(ss.probability(p), .0216)
ss = setset([s2])
self.assertAlmostEqual(ss.probability(p), .0096)
ss = setset([s12, s13])
self.assertAlmostEqual(ss.probability(p), .1368)
ss = setset([s1234])
self.assertAlmostEqual(ss.probability(p), .3024)
ss = setset([s0, s1, s2, s12, s13, s1234])
self.assertAlmostEqual(ss.probability(p), .4728)
def test_constructors(self):
ss = setset()
self.assertTrue(isinstance(ss, setset))
self.assertEqual(repr(ss), 'setset([])')
ss = setset([s0, s12, s13])
self.assertEqual(
repr(ss), "setset([set([]), set(['1', '2']), set(['1', '3'])])")
ss = setset([frozenset(s0), frozenset(s12), frozenset(s13)])
self.assertEqual(
repr(ss), "setset([set([]), set(['1', '2']), set(['1', '3'])])")
ss = setset({'include': list(s12), 'exclude': list(s4)})
self.assertEqual(repr(ss),
"setset([set(['1', '2']), set(['1', '2', '3'])])")
self.assertRaises(TypeError, setset, {'x': list(s1)})
self.assertRaises(TypeError, setset, {
'include': list(s12),
'exclude': list(s23)
})
# copy constructor
ss1 = setset([s0, s12, s13])
ss2 = ss1.copy()
self.assertTrue(isinstance(ss2, setset))
ss1.clear()
self.assertEqual(repr(ss1), 'setset([])')
self.assertEqual(
repr(ss2), "setset([set([]), set(['1', '2']), set(['1', '3'])])")
# repr
ss = setset({})
self.assertEqual(
repr(ss),
"setset([set([]), set(['1']), set(['2']), set(['3']), set(['4']), set(['1', ' ..."
)
def test_binary_operators(self):
u = [s0, s1, s12, s123, s1234, s134, s14, s4]
v = [s12, s14, s23, s34]
ss = setset(u) | setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u).union(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u)
ss |= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u)
ss.update(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u) & setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u).intersection(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u)
ss &= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u)
ss.intersection_update(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u) - setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u).difference(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u)
ss -= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u)
ss.difference_update(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u) ^ setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
ss = setset(u).symmetric_difference(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
ss = setset(u)
ss ^= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
ss = setset(u)
ss.symmetric_difference_update(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
v = [s12]
ss = setset(u) / setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u).quotient(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u)
ss /= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u)
ss.quotient_update(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u) % setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
ss = setset(u).remainder(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
ss = setset(u)
ss %= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
ss = setset(u)
ss.remainder_update(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
v = [s12, s14, s23, s34]
ss = setset(u).join(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s12, s123, s124, s1234, s134, s14, s23, s234, s34]))
ss = setset(u).meet(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s12, s14, s2, s23, s3, s34, s4]))
v = [s12, s14, s23, s34]
ss = setset(u).subsets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s12, s14, s4]))
ss = setset(u).supersets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s123, s1234, s134, s14]))
ss = setset(u).non_subsets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s123, s1234, s134]))
ss = setset(u).non_supersets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s4]))
ss1 = setset([s0, s12, s13])
ss2 = ss1.supersets('1')
self.assertTrue(isinstance(ss2, setset))
self.assertEqual(ss2, setset([s12, s13]))
ss2 = ss1.non_supersets('2')
self.assertTrue(isinstance(ss2, setset))
self.assertEqual(ss2, setset([s0, s13]))
v = [s1, s12, s13]
ss = setset(v)
s = ss.choice()
self.assertTrue(s in ss)
self.assertEqual(len(ss), 3)
ss.clear()
self.assertRaises(KeyError, ss.choice)
def test_unary_operators(self):
ss = setset([s0, s1, s12, s123, s1234, s134, s14, s4])
self.assertTrue(isinstance(~ss, setset))
self.assertEqual(~ss, setset([s124, s13, s2, s23, s234, s24, s3, s34]))
self.assertTrue(isinstance(ss.smaller(3), setset))
self.assertEqual(ss.smaller(3), setset([s0, s1, s12, s14, s4]))
self.assertTrue(isinstance(ss.larger(3), setset))
self.assertEqual(ss.larger(3), setset([s1234]))
self.assertTrue(isinstance(ss.set_size(3), setset))
self.assertEqual(ss.set_size(3), setset([s123, s134]))
self.assertTrue(isinstance(ss.len(3), setset))
self.assertEqual(ss.len(3), setset([s123, s134]))
ss = setset([s12, s123, s234])
self.assertTrue(isinstance(ss.minimal(), setset))
self.assertEqual(ss.minimal(), setset([s12, s234]))
self.assertTrue(isinstance(ss.maximal(), setset))
self.assertEqual(ss.maximal(), setset([s123, s234]))
ss = setset([s12, s14, s23, s34])
self.assertTrue(isinstance(ss.hitting(), setset))
self.assertEqual(
ss.hitting(), setset([s123, s1234, s124, s13, s134, s234, s24]))
def test_modifiers(self):
v = [s0, s12, s13]
ss = setset(v)
ss.add(s1)
self.assertTrue(s1 in ss)
ss.remove(s1)
self.assertTrue(s1 not in ss)
self.assertRaises(KeyError, ss.remove, s1)
ss.add(s0)
ss.discard(s0)
self.assertTrue(s0 not in ss)
ss.discard(s0) # no exception raised
ss = setset(v)
ss.add('2')
self.assertEqual(ss, setset([s12, s123, s2]))
ss = setset(v)
ss.remove('2')
self.assertEqual(ss, setset([s0, s1, s13]))
self.assertRaises(KeyError, ss.remove, '4')
ss = setset(v)
ss.discard('2')
self.assertEqual(ss, setset([s0, s1, s13]))
ss.discard('4') # no exception raised
v = [s1, s12, s13]
ss = setset(v)
s = ss.pop()
self.assertTrue(s not in ss)
self.assertEqual(ss | setset([s]), setset(v))
self.assertTrue(ss)
ss.clear()
self.assertFalse(ss)
self.assertRaises(KeyError, ss.pop)
u = [s0, s1, s12, s123, s1234, s134, s14, s4]
ss = setset(u)
ss.flip('1')
self.assertEqual(ss, setset([s0, s1, s14, s2, s23, s234, s34, s4]))
ss = setset(u)
ss.flip()
self.assertEqual(ss, setset([s0, s123, s1234, s2, s23, s234, s34, s4]))
def test_lookup(self):
ss1 = setset([s0, s12, s13])
self.assertTrue(s12 in ss1)
self.assertTrue(s1 not in ss1)
def test_comparison(self):
ss = setset([s12])
self.assertEqual(ss, setset([s12]))
self.assertNotEqual(ss, setset([s13]))
# __nonzero__
self.assertTrue(ss)
self.assertFalse(setset())
v = [s0, s12, s13]
ss = setset(v)
self.assertTrue(ss.isdisjoint(setset([s1, s123])))
self.assertFalse(ss.isdisjoint(setset([s1, s12])))
self.assertTrue(ss.issubset(setset(v)))
self.assertFalse(ss.issubset(setset([s0, s12])))
self.assertTrue(ss <= setset(v))
self.assertFalse(ss <= setset([s0, s12]))
self.assertTrue(ss < setset([s0, s1, s12, s13]))
self.assertFalse(ss < setset(v))
self.assertTrue(ss.issuperset(setset(v)))
self.assertFalse(ss.issuperset(setset([s1, s12])))
self.assertTrue(ss >= setset(v))
self.assertFalse(ss >= setset([s1, s12]))
self.assertTrue(ss > setset([set(), s12]))
self.assertFalse(ss > setset(v))
def test_modifiers(self):
v = [s0, s12, s13]
ss = setset(v)
ss.add(s1)
self.assertTrue(s1 in ss)
ss.remove(s1)
self.assertTrue(s1 not in ss)
self.assertRaises(KeyError, ss.remove, s1)
ss.add(s0)
ss.discard(s0)
self.assertTrue(s0 not in ss)
ss.discard(s0) # no exception raised
ss = setset(v)
ss.add('2')
self.assertEqual(ss, setset([s12, s123, s2]))
ss = setset(v)
ss.remove('2')
self.assertEqual(ss, setset([s0, s1, s13]))
self.assertRaises(KeyError, ss.remove, '4')
ss = setset(v)
ss.discard('2')
self.assertEqual(ss, setset([s0, s1, s13]))
ss.discard('4') # no exception raised
v = [s1, s12, s13]
ss = setset(v)
s = ss.pop()
self.assertTrue(s not in ss)
self.assertEqual(ss | setset([s]), setset(v))
self.assertTrue(ss)
ss.clear()
self.assertFalse(ss)
self.assertRaises(KeyError, ss.pop)
u = [s0, s1, s12, s123, s1234, s134, s14, s4]
ss = setset(u)
ss.flip('1')
self.assertEqual(ss, setset([s0, s1, s14, s2, s23, s234, s34, s4]))
ss = setset(u)
ss.flip()
self.assertEqual(ss, setset([s0, s123, s1234, s2, s23, s234, s34, s4]))
def test_lookup(self):
ss1 = setset([s0, s12, s13])
self.assertTrue(s12 in ss1)
self.assertTrue(s1 not in ss1)
def test_binary_operators(self):
u = [s0, s1, s12, s123, s1234, s134, s14, s4]
v = [s12, s14, s23, s34]
ss = setset(u) | setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u).union(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u)
ss |= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u)
ss.update(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s0, s1, s12, s123, s1234, s134, s14, s23, s34, s4]))
ss = setset(u) & setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u).intersection(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u)
ss &= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u)
ss.intersection_update(setset(u), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s14]))
ss = setset(u) - setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u).difference(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u)
ss -= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u)
ss.difference_update(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s4]))
ss = setset(u) ^ setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
ss = setset(u).symmetric_difference(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
ss = setset(u)
ss ^= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
ss = setset(u)
ss.symmetric_difference_update(setset(), setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s123, s1234, s134, s23, s34, s4]))
v = [s12]
ss = setset(u) / setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u).quotient(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u)
ss /= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u)
ss.quotient_update(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s3, s34]))
ss = setset(u) % setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
ss = setset(u).remainder(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
ss = setset(u)
ss %= setset(v)
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
ss = setset(u)
ss.remainder_update(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s134, s14, s4]))
v = [s12, s14, s23, s34]
ss = setset(u).join(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(
ss, setset([s12, s123, s124, s1234, s134, s14, s23, s234, s34]))
ss = setset(u).meet(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s12, s14, s2, s23, s3, s34, s4]))
v = [s12, s14, s23, s34]
ss = setset(u).subsets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s12, s14, s4]))
ss = setset(u).supersets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s12, s123, s1234, s134, s14]))
ss = setset(u).non_subsets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s123, s1234, s134]))
ss = setset(u).non_supersets(setset(v))
self.assertTrue(isinstance(ss, setset))
self.assertEqual(ss, setset([s0, s1, s4]))
ss1 = setset([s0, s12, s13])
ss2 = ss1.supersets('1')
self.assertTrue(isinstance(ss2, setset))
self.assertEqual(ss2, setset([s12, s13]))
ss2 = ss1.non_supersets('2')
self.assertTrue(isinstance(ss2, setset))
self.assertEqual(ss2, setset([s0, s13]))
v = [s1, s12, s13]
ss = setset(v)
s = ss.choice()
self.assertTrue(s in ss)
self.assertEqual(len(ss), 3)
ss.clear()
self.assertRaises(KeyError, ss.choice)
def test_comparison(self):
ss = setset([s12])
self.assertEqual(ss, setset([s12]))
self.assertNotEqual(ss, setset([s13]))
# __nonzero__
self.assertTrue(ss)
self.assertFalse(setset())
v = [s0, s12, s13]
ss = setset(v)
self.assertTrue(ss.isdisjoint(setset([s1, s123])))
self.assertFalse(ss.isdisjoint(setset([s1, s12])))
self.assertTrue(ss.issubset(setset(v)))
self.assertFalse(ss.issubset(setset([s0, s12])))
self.assertTrue(ss <= setset(v))
self.assertFalse(ss <= setset([s0, s12]))
self.assertTrue(ss < setset([s0, s1, s12, s13]))
self.assertFalse(ss < setset(v))
self.assertTrue(ss.issuperset(setset(v)))
self.assertFalse(ss.issuperset(setset([s1, s12])))
self.assertTrue(ss >= setset(v))
self.assertFalse(ss >= setset([s1, s12]))
self.assertTrue(ss > setset([set(), s12]))
self.assertFalse(ss > setset(v))
def test_unary_operators(self):
ss = setset([s0, s1, s12, s123, s1234, s134, s14, s4])
self.assertTrue(isinstance(~ss, setset))
self.assertEqual(~ss, setset([s124, s13, s2, s23, s234, s24, s3, s34]))
self.assertTrue(isinstance(ss.smaller(3), setset))
self.assertEqual(ss.smaller(3), setset([s0, s1, s12, s14, s4]))
self.assertTrue(isinstance(ss.larger(3), setset))
self.assertEqual(ss.larger(3), setset([s1234]))
self.assertTrue(isinstance(ss.set_size(3), setset))
self.assertEqual(ss.set_size(3), setset([s123, s134]))
self.assertTrue(isinstance(ss.len(3), setset))
self.assertEqual(ss.len(3), setset([s123, s134]))
ss = setset([s12, s123, s234])
self.assertTrue(isinstance(ss.minimal(), setset))
self.assertEqual(ss.minimal(), setset([s12, s234]))
self.assertTrue(isinstance(ss.maximal(), setset))
self.assertEqual(ss.maximal(), setset([s123, s234]))
ss = setset([s12, s14, s23, s34])
self.assertTrue(isinstance(ss.hitting(), setset))
self.assertEqual(ss.hitting(),
setset([s123, s1234, s124, s13, s134, s234, s24]))
