def test_1row_input(self): input = {'a': 1, 'b': 1} output = process_output_values(input) assert type(output) == list assert len(output) == 1 assert_frame_equal(output[0], DataFrame(input, index=[0]))
def test_np_1d_input(self): input = {'a': [1, 2, 3], 'b': array([3, 4, 5])} output = process_output_values(input) assert type(output) == list assert len(output) == 1 assert_frame_equal(output[0], DataFrame(input))
def test_list_input(self): input = [1, 2, 3] output = process_output_values(input) assert type(output) == list assert len(output) == 3 assert output == input
def test_dataframe_input(self): input = DataFrame({'a': [1, 2, 3]}) output = process_output_values(input) assert type(output) == list assert len(output) == 1 assert output == [input]
def test_np_2d_input(self): input = {'a': [1, 2, 3], 'b': array([[3, 4], [5, 5], [7, 6]])} output = process_output_values(input) expected_output = DataFrame({ 'a': [1, 2, 3], 'b': [array([3, 4]), array([5, 5]), array([7, 6])] }) assert type(output) == list assert len(output) == 1 assert_frame_equal(output[0], expected_output)
def test_none_input(self): input = None output = process_output_values(input) assert output == input