diff --git a/tests/test_indian.py b/tests/test_indian.py index 5b76f2439d01c0057a1b184ecb37ac1e83733031..1a179bd81dd275eb31bbe5f206f71d50b9c4a57b 100644 --- a/tests/test_indian.py +++ b/tests/test_indian.py @@ -22,17 +22,17 @@ def id_function(val): @pytest.mark.parametrize( "split,lenghts", [ - ("default", dict(train=83, validation=20, test=52)), - ("fold-0", dict(train=111, validation=28, test=16)), - ("fold-1", dict(train=111, validation=28, test=16)), - ("fold-2", dict(train=111, validation=28, test=16)), - ("fold-3", dict(train=111, validation=28, test=16)), - ("fold-4", dict(train=111, validation=28, test=16)), - ("fold-5", dict(train=112, validation=28, test=15)), - ("fold-6", dict(train=112, validation=28, test=15)), - ("fold-7", dict(train=112, validation=28, test=15)), - ("fold-8", dict(train=112, validation=28, test=15)), - ("fold-9", dict(train=112, validation=28, test=15)), + ("default", dict(train=84, validation=20, test=52)), + ("fold-0", dict(train=111, validation=28, test=17)), + ("fold-1", dict(train=112, validation=28, test=16)), + ("fold-2", dict(train=111, validation=29, test=16)), + ("fold-3", dict(train=112, validation=28, test=16)), + ("fold-4", dict(train=111, validation=29, test=16)), + ("fold-5", dict(train=113, validation=28, test=15)), + ("fold-6", dict(train=112, validation=29, test=15)), + ("fold-7", dict(train=113, validation=28, test=15)), + ("fold-8", dict(train=112, validation=29, test=15)), + ("fold-9", dict(train=113, validation=28, test=15)), ], ids=id_function, # just changes how pytest prints it )