def test_oss_sample_wrong_X(): """Test either if an error is raised when X is different at fitting and sampling""" # Create the object oss = OneSidedSelection(random_state=RND_SEED) oss.fit(X, Y) assert_raises(RuntimeError, oss.sample, np.random.random((100, 40)), np.array([0] * 50 + [1] * 50))
def test_oss_fit(): """Test the fitting method""" # Create the object oss = OneSidedSelection(random_state=RND_SEED) # Fit the data oss.fit(X, Y) # Check if the data information have been computed assert_equal(oss.min_c_, 0) assert_equal(oss.maj_c_, 1) assert_equal(oss.stats_c_[0], 6) assert_equal(oss.stats_c_[1], 9)