def test_groupby1(self):
df = get_iris_randomgroup()
enc_out = label_encoder(df, input_col='species', group_by=['random_group1', 'random_group2'])
print(enc_out['out_table'])
print(enc_out['model'].keys())
model_out = label_encoder_model(df, enc_out['model'])
print(model_out['out_table'])
def test(self):
le = label_encoder(self.testdata, input_col='species')
DF1 = le['out_table'].values
# print(DF1)
np.testing.assert_equal(DF1[0][5], 0)
np.testing.assert_equal(DF1[1][5], 0)
np.testing.assert_equal(DF1[2][5], 0)
np.testing.assert_equal(DF1[3][5], 0)
np.testing.assert_equal(DF1[4][5], 0)
le_model = label_encoder_model(self.testdata,
model=le['model'],
new_column_name='new_col')
DF2 = le_model['out_table'].values
# print(DF2)
np.testing.assert_equal(DF2[0][5], 0)
np.testing.assert_equal(DF2[1][5], 0)
np.testing.assert_equal(DF2[2][5], 0)
np.testing.assert_equal(DF2[3][5], 0)
np.testing.assert_equal(DF2[4][5], 0)