def test_partition_zero(self): self.assertEqual(main.partition(0, 2), 0) self.assertEqual(main.partition(2, 0), 0)
def test_partition_general(self): self.assertEqual(main.partition(6, 3), 3) self.assertEqual(main.partition(7, 3), 4) self.assertEqual(main.partition(10, 2), 5) self.assertEqual(main.partition(2, 2), 1) self.assertEqual(main.partition(10, 8), 2)
def test_partition_k_greater(self): self.assertEqual(main.partition(5, 6), 0)