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)