def test_data_formats(self):
"""
Test the data format options.
"""
for input_file in self.input_file_names:
input_data_obj = Data(input_file['path'])
self.assertIsInstance(input_data_obj.data, pd.DataFrame)
input_data_obj.data_format = "records"
self.assertIsInstance(input_data_obj.data, list)
with self.assertRaises(ValueError) as exc:
input_data_obj.data_format = "NON_EXISTENT"
self.assertEqual(
str(exc.exception),
"The data format must be one of the following: " +
"['dataframe', 'records']")
def test_allowed_data_formats(self):
"""
Determine if the csv file data_formats can be used
"""
for input_file in self.file_or_buf_list:
input_data_obj = Data(input_file['path'])
for data_format in list(input_data_obj._data_formats.keys()):
input_data_obj.data_format = data_format
self.assertEqual(input_data_obj.data_format, data_format)
data = input_data_obj.data
if data_format == "dataframe":
import pandas as pd
self.assertIsInstance(data, pd.DataFrame)
elif data_format in ["records", "json"]:
self.assertIsInstance(data, list)
self.assertIsInstance(data[0], str)
