import re
[docs]def get_first_line(output):
match = re.search("^(.*)$", output, flags=re.MULTILINE)
if match is None:
return None
else:
return match.group(1)
[docs]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