def test_rbf():
"""Assert that the TPOT RBFSampler outputs the input dataframe when # of
training features is 0"""
tpot_obj = TPOT()
assert np.array_equal(tpot_obj._rbf(training_testing_data.ix[:, -3:], 0.1),
training_testing_data.ix[:, -3:])
def test_rbf():
"""Assert that the TPOT RBFSampler outputs the input dataframe when # of
training features is 0"""
tpot_obj = TPOT()
assert np.array_equal(tpot_obj._rbf(training_testing_data.ix[:, -3:], 0.1),
training_testing_data.ix[:, -3:])
def test_rbf_2():
"""Assert that RBF returns an object of the same type as the input dataframe
Also assert that the number of rows is identical between the input dataframe
and output dataframe.
"""
tpot_obj = TPOT()
input_df = training_testing_data
output_df = tpot_obj._rbf(input_df, 0.1)
assert type(input_df) == type(output_df)
(in_rows, in_cols) = input_df.shape
(out_rows, out_cols) = output_df.shape
assert in_rows == out_rows
def test_rbf_2():
"""Assert that RBF returns an object of the same type as the input dataframe
Also assert that the number of rows is identical between the input dataframe
and output dataframe.
"""
tpot_obj = TPOT()
input_df = training_testing_data
output_df = tpot_obj._rbf(input_df, 0.1)
assert type(input_df) == type(output_df)
(in_rows, in_cols) = input_df.shape
(out_rows, out_cols) = output_df.shape
assert in_rows == out_rows
