def test_from_file_multiple_types(mock_file, multiple_types_csv): path = 'path/to/file' processor_from_file = CSVProcessor.from_file(path) assert processor_from_file == CSVProcessor(multiple_types_csv) mock_file.assert_called_once_with(path)
def test_from_file(mock_file, simple_int_csv): path = 'path/to/file' processor_from_file = CSVProcessor.from_file(path) assert processor_from_file == CSVProcessor(simple_int_csv) mock_file.assert_called_once_with(path)
def test_from_file(mock_file, simple_int_csv): path = 'path/to/file' processor_from_file = CSVProcessor.from_file(path) assert processor_from_file == CSVProcessor(simple_int_csv)
def test_from_file_multi(mock_file, multiple_types_csv): path = 'path/to/file' processor_from_file = CSVProcessor.from_file(path) assert processor_from_file == CSVProcessor(multiple_types_csv)
from csv_processor import CSVProcessor if __name__ == '__main__': csv = CSVProcessor.from_file('sample.csv', types=(int, str, str, int)) print(csv.csv) print(csv.get_column(1))