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])