Files
galaxy/test/base/asserts/tabular.py
T
John Chilton 8814d94e60 PEP-8 fixes for test.base.asserts.
Some of the first Python code I ever wrote - and one can tell :).
2014-11-25 20:57:56 -05:00

20 lines
634 B
Python

import re
def get_first_line(output):
match = re.search("^(.*)$", output, flags=re.MULTILINE)
if match is None:
return None
else:
return match.group(1)
def assert_has_n_columns(output, n, sep='\t'):
""" Asserts the tabular output contains n columns. The optional
sep argument specifies the column seperator used to determine the
number of columns."""
n = int(n)
first_line = get_first_line(output)
assert first_line is not None, "Was expecting output with %d columns, but output was empty." % n
assert len(first_line.split(sep)) == n, "Output does not have %d columns." % n