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)