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