def test_exceptions_key(self):
fvm = FeatureVectorMatrix()
caught = False
try:
fvm.add_row(v3, 'a')
except IndexError:
caught = True
self.assertTrue(caught, "You can't add a list to a fvm with no columns defined")
fvm.set_column_names(c3)
self.assertEquals(3, fvm.column_count())
fvm.add_row(v3, 'a')
self.assertEquals(1, fvm.row_count())
fvm.add_row(v3, 'a')
self.assertEquals(1, fvm.row_count())
caught = False
try:
fvm.add_row(v4, 'b')
except IndexError:
caught = True
self.assertTrue(caught, "You can't add a list to a fvm with more columns than the given list")
fvm.add_row(d4, 'c')
self.assertEquals(len(fvm.column_names()), 4)
self.assertEquals(2, fvm.row_count())
# test that the already added row gets the extra 0 appended
self.assertEquals(fvm.get_row_list(0), [1, 2, 3,0])
def test_matrix(self):
fvm = FeatureVectorMatrix()
fvm.set_column_names(c3)
fvm.add_row(d3, 'one')
fvm.add_row(d4, 'two')
fvm.add_row(d4, 'three')
self.assertEquals(4, fvm.column_count())
self.assertEquals(3, fvm.row_count())
v3_old = fvm.get_row_list(0)
self.assertEquals(fvm.get_row_dict('two'), d4)
rfvm = fvm.transpose()
self.assertEquals(3, rfvm.column_count())
self.assertEquals(4, rfvm.row_count())
self.assertEquals(rfvm.get_row_dict(2), {'three': 3, 'two': 3, 'one': 3})
self.assertEquals(rfvm.get_row_dict(3), {'two': 4, 'three': 4})
self.assertEquals(rfvm.column_names(), ['one', 'two', 'three'])
self.assertEquals(rfvm.get_row_list(2), [3, 3, 3])
self.assertEquals(rfvm.get_row_list(3), [0, 4, 4])
fvm2 = rfvm.transpose()
self.assertEquals(4, fvm2.column_count())
self.assertEquals(3, fvm2.row_count())
self.assertEquals(fvm2.get_row_dict(2), d4)
self.assertEquals(fvm2.get_row_dict(0), d3)
self.assertEquals(fvm2.get_row_list(0), v3_old)
self.assertEquals(fvm2.get_row_dict('two'), d4)
def test_exceptions(self):
fvm = FeatureVectorMatrix()
caught = False
try:
fvm.add_row(v3)
except IndexError:
caught = True
self.assertTrue(caught, "You can't add a list to a fvm with no columns defined")
fvm.set_column_names(c3)
self.assertEquals(3, fvm.column_count())
fvm.add_row(v3)
self.assertEquals(1, fvm.row_count())
caught = False
try:
print fvm.keys()
except NotImplementedError:
caught = True
self.assertTrue(caught, "You can't get keys without all rows keyed")
caught = False
try:
fvm.transpose()
except NotImplementedError:
caught = True
self.assertTrue(caught, "You can't rotate a fvm without all rows keyed")
caught = False
try:
fvm.add_row(v4)
except IndexError:
caught = True
self.assertTrue(caught, "You can't add a list to a fvm with more columns than the given list")
fvm.add_row(d4)
self.assertEquals(len(fvm.column_names()), 4)
# test that the already added row gets the extra 0 appended
self.assertEquals(fvm.get_row_list(0), [1, 2, 3,0])