def test_unknown_type_column(self):
data = np.array(self.header0)
creator = _TableBuilder._get_column_creator("")
column = creator(data, 0)
self.assertIsNone(column.valuemap)
np.testing.assert_array_equal(column.values, [0.1, 0.2, 0])
np.testing.assert_array_equal(column.orig_values, ["0.1", "0.2", "0.0"])
self.assertEqual(column.coltype, ContinuousVariable)
self.assertDictEqual(column.coltype_kwargs, {})
def test_string_column(self):
data = np.array(self.header1[1:])
creator = _TableBuilder._get_column_creator("s")
column = creator(data, 0)
self.assertIsNone(column.valuemap)
np.testing.assert_array_equal(column.values, ["red", "red", "green"])
np.testing.assert_array_equal(column.orig_values,
["red", "red", "green"])
self.assertEqual(column.coltype, StringVariable)
self.assertDictEqual(column.coltype_kwargs, {})
def test_discrete_column(self):
data = np.array(self.header1[1:])
creator = _TableBuilder._get_column_creator("d")
columns = creator(data, 0)
self.assertListEqual(columns.valuemap, ['green', 'red'])
np.testing.assert_array_equal(columns.values, ["red", "red", "green"])
np.testing.assert_array_equal(columns.orig_values,
["red", "red", "green"])
self.assertEqual(columns.coltype, DiscreteVariable)
self.assertDictEqual(columns.coltype_kwargs, {})
def test_time_column(self):
data = np.array(self.header1[1:])
creator = _TableBuilder._get_column_creator("t")
column = creator(data, 1)
self.assertIsNone(column.valuemap)
np.testing.assert_array_equal(
column.values, ['2019-10-10', '2019-10-12', '2019-10-11'])
np.testing.assert_array_equal(
column.orig_values, ['2019-10-10', '2019-10-12', '2019-10-11'])
self.assertEqual(column.coltype, TimeVariable)
self.assertDictEqual(column.coltype_kwargs, {})
def test_column_parts_discrete_values(self):
data = np.array(self.header1[1:])
vals = "green red"
creator = _TableBuilder._get_column_creator(vals)
column = creator(data, 0, vals)
self.assertListEqual(column.valuemap, ['green', 'red'])
np.testing.assert_array_equal(column.values, ["red", "red", "green"])
np.testing.assert_array_equal(column.orig_values,
["red", "red", "green"])
self.assertEqual(column.coltype, DiscreteVariable)
self.assertDictEqual(column.coltype_kwargs, {})
def test_continuous_column_raises(self):
data = np.array([["a", "2"], ["3", "4"]])
creator = _TableBuilder._get_column_creator("c")
self.assertRaises(ValueError, creator, data, 0)
