class SubsetsTest(TestCase):
def setUp(self):
self.l = [0, 3, 4]
self.pcset = PCSet(self.l)
def testSubsets(self):
subs = [sub for sub in self.pcset.subsets()]
self.assertEqual(subs, [
[3, 4], [4], [], [3], [],
[0, 4], [4], [], [0], [],
[0, 3], [3], [], [0], []
])
self.assertTrue(isinstance(subs[0], PCSet))
def testSubPrimes(self):
subprimes = [subprime for subprime in self.pcset.subprimes()]
self.assertEqual(subprimes, [
[0, 1], [0], [], [0], [],
[0, 4], [0], [], [0], [],
[0, 3], [0], [], [0], []
])
