def test_split_size_bound():
with pytest.raises(ValueError):
pre.splitter(TOY_X, target_index=1, split_size=1.5, seed=1)
pre.splitter(TOY_X, target_index=1, split_size=1.0, seed=1)
pre.splitter(TOY_X, target_index=1, split_size=0.0, seed=1)
pre.splitter(TOY_X, target_index=1, split_size=-1.5, seed=1)
def test_split_proportions_size():
X_train, y_train, X_test, y_test = pre.splitter(TOY_X, target_index=1, split_size=0.25, seed=1)
assert(X_train.shape[0] == 3), "size of X train doesn't match"
def test_X_size():
with pytest.raises(ValueError):
pre.splitter(np.array([1]), target_index=1, split_size=0.2, seed=1)
pre.splitter(np.array([]), target_index=1, split_size=0.2, seed=1)
def test_seed_type():
with pytest.raises(TypeError):
pre.splitter(TOY_X, target_index=1, split_size=0.2, seed="1")
pre.splitter(TOY_X, target_index=1, split_size=0.2, seed=0.5)
def test_splitsize_type():
with pytest.raises(TypeError):
pre.splitter(TOY_X, target_index=1, split_size="0.2", seed=1)
pre.splitter(TOY_X, target_index=1, split_size=5, seed=1)
def test_target_index_type():
with pytest.raises(TypeError):
pre.splitter(TOY_X, target_index="1", split_size=0.2, seed=1)
pre.splitter(TOY_X, target_index=3.5, split_size=0.2, seed=1)
def test_X_type():
with pytest.raises(TypeError):
pre.splitter("X", target_index=1, split_size=0.2, seed=1)
pre.splitter({1}, target_index=1, split_size=0.2, seed=1)
pre.splitter(3.5, target_index=1, split_size=0.2, seed=1)