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)
