def test_read_data_returns_correct_results():
args = {
"filename": "data/ex4.dat",
"title": None,
"width": 50,
"format": "{:<5.2f}",
"suffix": "",
"no_labels": False,
"color": None,
"vertical": False,
"stacked": False,
"different_scale": False,
"calendar": False,
"start_dt": None,
"custom_tick": "",
"delim": "",
"verbose": False,
"version": False,
}
categories, labels, data, colors = tg.read_data(args)
assert categories == ["Boys", "Girls"]
assert labels == ["2007", "2008", "2009", "2010", "2011", "2012", "2014"]
assert data == [
[183.32, 190.52],
[231.23, 5.0],
[16.43, 53.1],
[50.21, 7.0],
[508.97, 10.45],
[212.05, 20.2],
[30.0, 20.0],
]
assert colors == []
def test_read_data_verbose():
with patch("sys.stdout", new=StringIO()) as output:
args = {
"filename": "data/ex1.dat",
"title": None,
"width": 50,
"format": "{:<5.2f}",
"suffix": "",
"no_labels": False,
"color": None,
"vertical": False,
"stacked": False,
"different_scale": False,
"calendar": False,
"start_dt": None,
"custom_tick": "",
"delim": "",
"verbose": True,
"version": False,
}
tg.read_data(args)
output = output.getvalue().strip()
assert output == ">> Reading data from data/ex1.dat"
def test_read_data_with_title_prints_title():
with patch("sys.stdout", new=StringIO()) as output:
args = {
"filename": "data/ex4.dat",
"title": "spaghetti",
"width": 50,
"format": "{:<5.2f}",
"suffix": "",
"no_labels": False,
"color": None,
"vertical": False,
"stacked": False,
"different_scale": False,
"calendar": False,
"start_dt": None,
"custom_tick": "",
"delim": "",
"verbose": False,
"version": False,
}
tg.read_data(args)
output = output.getvalue().strip()
assert output == "# spaghetti\n\n▇ Boys ▇ Girls"