def setUp(self): items = [Item(1, 1), Item(10, 1), Item(20, 2), Item(30, 3), Item(40, 4), Item(50, 5), Item(50, 6), Item(40, 7), Item(30, 8), Item(20, 9)] self.solved = [items[0], items[1], items[2], items[3], items[6], items[7], items[8], items[9]] self.solution = KnapSack(202) self.solution.solve(items)
class KnapSackTest(unittest.TestCase): def setUp(self): items = [Item(1, 1), Item(10, 1), Item(20, 2), Item(30, 3), Item(40, 4), Item(50, 5), Item(50, 6), Item(40, 7), Item(30, 8), Item(20, 9)] self.solved = [items[0], items[1], items[2], items[3], items[6], items[7], items[8], items[9]] self.solution = KnapSack(202) self.solution.solve(items) def test_knapsack_simple(self): self.assertTrue(sorted(self.solution.items) == sorted(self.solved)) def test_knapsack_weight(self): self.assertEqual(self.solution.weight, 201) def test_knapsack_value(self): self.assertEqual(self.solution.value, 37)