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)