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], [] ])