def test_sort_stack(self): TEST_DATA = [ Stack([1]), Stack([1, 2, 3, 4]), Stack([4, 3, 2, 1]), Stack([1, 3, 2, 4, 7, 4, 5, 8, 6, 9])] for test in TEST_DATA: sort_stack(test) self.assertTrue(self.is_sorted(test))
def test_sort_stack(self): stack = Stack() stack.push(4) stack.push(5) stack.push(2) stack.push(7) sorted_stack = sort_stack(stack) self.assertEqual(sorted_stack.pop(), 2) self.assertEqual(sorted_stack.pop(), 4) self.assertEqual(sorted_stack.pop(), 5) self.assertEqual(sorted_stack.pop(), 7)
def test_generic(self): for idx, stack in enumerate(self.stacks): self.assertEqual(self.sorted_stacks[idx], sort_stack(stack))