Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_read_data_with_title_prints_title(self):
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'
def test_read_data_returns_correct_results(self):
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(self):
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'