def math_binary_deserialize_add_test(self):
math_binary_tf = MathBinary(input_features=['a', 'b'], output_features='a_plus_b', transform_type='add')
Xres = math_binary_tf.fit_transform(self.df[['a', 'b']])
assert_frame_equal(pd.DataFrame(self.df.a + self.df.b, columns=['a']), Xres)
math_binary_tf.serialize_to_bundle(self.tmp_dir, math_binary_tf.name)
node_name = "{}.node".format(math_binary_tf.name)
math_binary_ds_tf = MathBinary()
math_binary_ds_tf = math_binary_ds_tf.deserialize_from_bundle(self.tmp_dir, node_name)
res_a = math_binary_tf.transform(self.df[['a', 'b']])
res_b = math_binary_ds_tf.transform(self.df[['a', 'b']])
assert_frame_equal(res_a, res_b)
def math_binary_deserialize_exp_test(self):
math_binary_tf = MathBinary(input_features=['a', 'b'], output_features=['a_plus_b'], transform_type='add')
Xres = math_binary_tf.fit_transform(self.df[['a', 'b']])
self.assertEqual( self.df.a[0] + self.df.b[0], Xres[0])
math_binary_tf.serialize_to_bundle(self.tmp_dir, math_binary_tf.name)
node_name = "{}.node".format(math_binary_tf.name)
math_binary_ds_tf = MathBinary()
math_binary_ds_tf = math_binary_ds_tf.deserialize_from_bundle(self.tmp_dir, node_name)
res_a = math_binary_tf.transform(self.df[['a', 'b']])
res_b = math_binary_ds_tf.transform(self.df[['a', 'b']])
self.assertEqual(res_a[0], res_b[0])