class ReprTest(TestCase):
"""PCSet and PSet repr's vary based on ordered and multiset settings"""
def setUp(self):
self.l = [0, 1, -7, 16, 1, 0]
self.pcset = PCSet(self.l)
self.pset = PSet(self.l)
def testreprO(self):
self.pcset.ordered(True)
self.pset.ordered(True)
self.pcset.multiset(False)
self.pset.multiset(False)
self.assertEqual(self.pcset.__repr__(), '[0, 1, 5, 4]')
self.assertEqual(self.pset.__repr__(), '[0, 1, -7, 16]')
def testreprO_Multi(self):
pcset = PCSet(self.l, ordered=True, multiset=True)
pset = PSet(self.l, ordered=True, multiset=True)
self.assertEqual(pcset.__repr__(), '[0, 1, 5, 4, 1, 0]')
self.assertEqual(pset.__repr__(), '[0, 1, -7, 16, 1, 0]')
def testreprUO(self):
self.pcset.ordered(False)
self.pset.ordered(False)
self.pcset.multiset(False)
self.pset.multiset(False)
self.assertEqual(self.pcset.__repr__(), '[0, 1, 4, 5]')
self.assertEqual(self.pset.__repr__(), '[-7, 0, 1, 16]')
def testreprUO_Multi(self):
pcset = PCSet(self.l, ordered=False, multiset=True)
pset = PSet(self.l, ordered=False, multiset=True)
self.assertEqual(pcset.__repr__(), '[0, 0, 1, 1, 4, 5]')
self.assertEqual(pset.__repr__(), '[-7, 0, 0, 1, 1, 16]')
class SettingsTest(TestCase):
"""PCSet and PSet instances have various methods to change settings"""
def setUp(self):
self.l = [0, 4, 9, 11]
self.pcset = PCSet(self.l)
self.pset = PSet(self.l)
def testMod(self):
self.pcset.mod(7)
self.pset.mod(13)
self.assertEqual(self.pcset._mod, 7)
self.assertEqual(self.pset._mod, 13)
def testModInvalid(self):
for n in [0, 32]:
self.assertRaises(self.pcset.InvalidModulus, self.pcset.mod, n)
def testCanon(self):
a = self.pcset
b = self.pset
a.canon(True, False, False)
b.canon(False, True, True)
self.assertTrue(
a._canon_t == True and a._canon_i == False and a._canon_m == False
)
self.assertTrue(
b._canon_t == False and b._canon_i == True and b._canon_m == True
)
def testDefault_m(self):
self.pcset.default_m(7)
self.assertEqual(self.pcset._default_m, 7)
def testOrdered(self):
self.pcset.ordered(True)
self.pset.ordered(True)
self.assertTrue(
self.pcset._ordered == True and self.pset._ordered == True
)
self.pcset.ordered(False)
self.pset.ordered(False)
self.assertTrue(
self.pcset._ordered == False and self.pset._ordered == False
)
def testMultiset(self):
self.pcset.multiset(True)
self.pset.multiset(True)
self.assertTrue(
self.pcset._multiset == True and self.pset._multiset == True
)
self.pcset.multiset(False)
self.pset.multiset(False)
self.assertTrue(
self.pcset._multiset == False and self.pset._multiset == False
)
class SequenceTest(TestCase):
"""PCSets and PSets behave like sequences with iteration and len()"""
def setUp(self):
self.l = [0, 1, 2, 4]
self.pcset = PCSet(self.l)
self.pset = PSet(self.l)
def testiterTest(self):
pcout = [pc for pc in self.pcset]
pout = [p for p in self.pset]
self.assertTrue(pcout == self.l and pout == self.l)
def testlenTest(self):
self.assertEqual(len(self.pcset), 4)
self.pset.multiset(False)
self.assertEqual(len(self.pset + [0, 2, 5]), 5)
self.pset.multiset(True)
self.assertEqual(len(self.pset + [0, 2, 5]), 7)
