def test_number_of_elements_for_matrix_in_data_is_more_than_width_mul_height(self): try: get_spiral_elements(3, 2, [10]) except AssertionError: self.assertTrue(True) except Exception: raise Exception, "Unhandled error if number of elements is more than width*height"
def test_square_matrix_3x3(self): width = height = 3 data = range(1, 10) expectation = [1, 2, 3, 6, 9, 8, 7, 4, 5] self.assertEqual(get_spiral_elements(width, height, data), expectation)
def test_square_matrix_2x2(self): width = height = 2 data = range(1, 5) expectation = [1, 2, 4, 3] self.assertEqual(get_spiral_elements(width, height, data), expectation)
def test_only_1_element(self): width = height = 1 data = [1] expectation = [1] self.assertEqual(get_spiral_elements(width, height, data), expectation)
def test_rectangle_matrix_2x3(self): width = 2 height = 3 data = range(6) expectation = [0, 1, 3, 5, 4, 2] self.assertEqual(get_spiral_elements(width, height, data), expectation)
def test_rectangle_matrix_3x2(self): width = 3 height = 2 data = range(6) expectation = [0, 1, 2, 5, 4, 3] self.assertEqual(get_spiral_elements(width, height, data), expectation)