def test_to_csv(self): data = BoxList([ { "Number": 1, "Name": "Chris", "Country": "US" }, { "Number": 2, "Name": "Sam", "Country": "US" }, { "Number": 3, "Name": "Jess", "Country": "US" }, { "Number": 4, "Name": "Frank", "Country": "UK" }, { "Number": 5, "Name": "Demo", "Country": "CA" }, ]) file = Path(tmp_dir, "csv_file.csv") data.to_csv(filename=file) assert file.read_text().startswith("Number,Name,Country\n1,Chris,US") assert data.to_csv().endswith( "2,Sam,US\r\n3,Jess,US\r\n4,Frank,UK\r\n5,Demo,CA\r\n")
def test_bad_csv(self): data = BoxList([ {'test': 1}, {'bad': 2, 'data': 3} ]) file = Path(tmp_dir, 'csv_file.csv') with pytest.raises(BoxError): data.to_csv(file)
def test_to_csv(self): data = BoxList([ {'Number': 1, 'Name': 'Chris', 'Country': 'US'}, {'Number': 2, 'Name': 'Sam', 'Country': 'US'}, {'Number': 3, 'Name': 'Jess', 'Country': 'US'}, {'Number': 4, 'Name': 'Frank', 'Country': 'UK'}, {'Number': 5, 'Name': 'Demo', 'Country': 'CA'}, ]) file = Path(tmp_dir, 'csv_file.csv') data.to_csv(file) assert file.read_text().startswith("Number,Name,Country\n1,Chris,US")
def test_bad_csv(self): data = BoxList([{"test": 1}, {"bad": 2, "data": 3}]) file = Path(tmp_dir, "csv_file.csv") with pytest.raises(BoxError): data.to_csv(file)