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