def test_groupby1(self): df = get_iris_randomgroup() enc_out = one_hot_encoder(df, input_cols=['sepal_length', 'sepal_width'], prefix_list=['a', 'b'], group_by=['random_group1', 'random_group2']) print(enc_out['out_table']) model_out = one_hot_encoder_model(df, enc_out['model']) print(model_out['out_table'])
def test(self): ohe = one_hot_encoder(self.testdata, input_cols=['species'], prefix_list=['list']) DF1 = ohe['out_table'].values # print(DF1) np.testing.assert_array_equal(DF1[0][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF1[1][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF1[2][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF1[3][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF1[4][5:8], [1.0, 0.0, 0.0]) ohe_model = one_hot_encoder_model(self.testdata, model=ohe['model']) DF2 = ohe_model['out_table'].values # print(DF2) np.testing.assert_array_equal(DF2[0][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF2[1][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF2[2][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF2[3][5:8], [1.0, 0.0, 0.0]) np.testing.assert_array_equal(DF2[4][5:8], [1.0, 0.0, 0.0])