def test_empty_table(self): data = [] data_table = DataTable(data) self.assertEqual(data_table.nrows, 0) self.assertEqual(data_table.ncols, 0) data_table.cleanup() data_table.serialize()
def test_clean_values(self): data = [[" a b ", 3.898989, None], [None, None, None]] data_table = DataTable(data) data_table.clean_values() self.assertEqual(data_table.rows[0][0], "a b") self.assertEqual(data_table.rows[0][1], Decimal("3.89")) data_table.trim_empty_rows() data_table.trim_empty_columns() self.assertEqual(data_table.serialize(), [["a b", Decimal("3.89")]]) data_table = DataTable(data) data_table.cleanup() self.assertEqual(data_table.serialize(), [["a b", Decimal("3.89")]])