def test_split_train_test(self, numeric_dataset):
"""Test split"""
n_split = 2
flow = Flow(seed=42)
flow = flow.from_source(Source(fin=numeric_dataset.open())).map(
lambda x: [float(_) for _ in x]
)
flow_left = Flow(flow=flow).split(n_split, 0)
flow_right = Flow(flow=flow).split(n_split, 1)
assert sum(flow_left.map(lambda x: 1).reduce(
lambda a, b: a + b
).eval() + flow_right.map(lambda x: 1).reduce(
lambda a, b: a + b
).eval()) == 10
