def test_read_raw_compressed():
input_file = 'test.tsv.gz'
with gzip.open(input_file, 'wt') as g:
g.write(test_input_data)
raw = reader.read_raw(input_file)
os.remove(input_file)
assert_frame_equal(raw, expected_raw)
def test_read_raw():
input_file = 'test.tsv'
with open(input_file, 'wt') as f:
f.write(test_input_data)
raw = reader.read_raw(input_file)
os.remove(input_file)
assert_frame_equal(raw, expected_raw)