def test_nystroem():
"""Assert that the TPOT Nystroem preprocessor outputs the input dataframe
when the number of training features is 0"""
tpot_obj = TPOT()
assert np.array_equal(tpot_obj._nystroem(training_testing_data.ix[:, -3:], 1, 0.1, 1),
training_testing_data.ix[:, -3:])
def test_nystroem():
"""Assert that the TPOT Nystroem preprocessor outputs the input dataframe
when the number of training features is 0"""
tpot_obj = TPOT()
assert np.array_equal(tpot_obj._nystroem(training_testing_data.ix[:, -3:], 1, 0.1, 1),
training_testing_data.ix[:, -3:])
def test_nystroem_2():
"""Assert that Nystroem returns the same object type as the input object type.
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._nystroem(input_df, 1, 0.1, 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_nystroem_2():
"""Assert that Nystroem returns the same object type as the input object type.
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._nystroem(input_df, 1, 0.1, 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
