def test_input_csv(self, read_csv):
"""Tests valid input CSV file."""
expected = self.input_df
read_csv.return_value = expected
f = _make_csv_tempfile(get_sample_image_csv_data())
actual = client.to_dataframe(f.name)
pd.testing.assert_frame_equal(actual, expected)
def test_input_image_dir(self, mock_fn):
"""Tests valid input image directory."""
mock_fn.return_value = self.input_df
with tempfile.TemporaryDirectory() as input_data:
actual = client.to_dataframe(input_data)
pd.testing.assert_frame_equal(actual, self.input_df)
def test_error_invalid_inputs(self):
"""Tests error handling with different invalid inputs."""
inputs = [0, 'not_a_csv_file', list(), dict()]
for input_data in inputs:
with self.assertRaises(ValueError):
client.to_dataframe(input_data)
def test_input_dataframe_with_header(self):
"""Tests valid input dataframe with header specified."""
names = list(self.input_df.columns[0:-1])
actual = client.to_dataframe(self.input_df, names=names)
pd.testing.assert_frame_equal(actual, self.input_df[names])
def test_input_dataframe_no_names_specified(self):
"""Tests valid input dataframe with no header names specified."""
actual = client.to_dataframe(self.input_df)
pd.testing.assert_frame_equal(actual, self.input_df)
