test_object_analyzer.py 文件源码

def test__replace_objects_by_integers(self):
        data = pd.DataFrame([{'column1': 1.3, 'column2': "Bla"},
                             {'column1': 3.2, 'column2': "Bla"},
                             {'column1': 2.7, 'column2': "Aha"}])
        data = metafeatures.core.object_analyzer \
            ._replace_objects_by_integers(data,
                                          {0: {'type': 'numerical',
                                               'name': 'column1',
                                               'is_target': False},
                                           1: {'type': 'categorical',
                                               'name': 'column2',
                                               'is_target': False}})
        print(data)
        self.assertEqual(data.dtypes[0], np.float)
        self.assertEqual(data.dtypes[1], np.float)
        np.testing.assert_allclose(data.iloc[:, 1], [0, 0, 1])