def test_valid_column_set_as_index(self): """ Test that the index is a valid column name """ column = np.random.choice(self.frame.columns.values) frame, mapping = rad.preprocess(self.frame, index=column) self.assertEqual(frame.index.name, column)
def test_populated_mapping_given_nonnumeric_array(self): """ Test that all string columns get mapped to its respective integer. """ shape = (len(self.frame), np.random.randint(1, 5)) nd_array = np.random.choice(["a", "b"], shape) frame, mapping = rad.preprocess(nd_array) self.assertEqual(len(mapping), nd_array.shape[1])
def test_empty_mapping_given_numeric_array(self): """ Test that if a numeric array is given, no mappings are returned. """ _, mapping = rad.preprocess(self.frame) self.assertEqual(len(mapping), 0)
def test_preprocess_doesnt_change_numeric_array(self): frame, _ = rad.preprocess(self.frame) # numeric arrays, after `preprocess`, are exactly the same as before. self.assertEqual(frame.values.all(), self.frame.values.all())