def test_read_categorical(self):
res = read_metadata(self.data2)
exp = pd.DataFrame([['a', 0.0, '1', 4.0], ['1', 1.0, 'a', 5.0],
['3', 4.0, 'c', 12.0]],
columns=['sampleid', 'col1', 'col2', 'col3'])
exp = exp.set_index('sampleid')
pdt.assert_frame_equal(exp, res)
def test_read_index(self):
res = read_metadata(self.data1)
exp = pd.DataFrame([['0', 0.0, 2.0, 4.0], ['1', 1.0, 3.0, 5.0],
['3', 4.0, 8.0, 12.0]],
columns=['sampleid', 'col1', 'col2', 'col3'])
exp = exp.set_index('sampleid')
pdt.assert_frame_equal(exp, res)