@@ -81,13 +81,13 @@ def matching_data():
8181
8282def test_aatest (aa_data ):
8383 mapping = {
84- "aa-casual" : AATest (n_iterations = 10 ),
84+ "aa-casual" : AATest (random_states = [ 56 , 72 , 2 , 43 ] ),
8585 "aa-rs" : AATest (random_states = [56 , 72 , 2 , 43 ]),
8686 "aa-strat" : AATest (stratification = True , random_states = [56 , 72 , 2 , 43 ]),
87- "aa-sample" : AATest (n_iterations = 10 , sample_size = 0.3 ),
88- "aa-cat_target" : AATest (n_iterations = 10 ),
89- "aa-equal_var" : AATest (n_iterations = 10 , t_test_equal_var = False ),
90- "aa-n" : AATest (n_iterations = 10 , groups_sizes = [0.5 , 0.2 , 0.3 ]),
87+ "aa-sample" : AATest (sample_size = 0.3 , random_states = [ 56 , 72 , 2 , 43 ] ),
88+ "aa-cat_target" : AATest (random_states = [ 56 , 72 , 2 , 43 ] ),
89+ "aa-equal_var" : AATest (equal_variance = False , random_states = [ 56 , 72 , 2 , 43 ] ),
90+ "aa-n" : AATest (groups_sizes = [0.5 , 0.2 , 0.3 ], random_states = [ 56 , 72 , 2 , 43 ]),
9191 }
9292
9393 mapping_resume = {
@@ -122,7 +122,7 @@ def test_aatest(aa_data):
122122 {
123123 "TTest aa test" : {0 : "OK" , 1 : "OK" },
124124 "KSTest aa test" : {0 : "OK" , 1 : "OK" },
125- "TTest best split" : {0 : "NOT OK" , 1 : "NOT OK" },
125+ "TTest best split" : {0 : "OK" , 1 : "OK" },
126126 "KSTest best split" : {0 : "OK" , 1 : "OK" },
127127 "result" : {0 : "OK" , 1 : "OK" },
128128 }
@@ -150,7 +150,7 @@ def test_aatest(aa_data):
150150 "aa-n" : pd .DataFrame (
151151 {
152152 "TTest aa test" : {0 : "OK" , 1 : "OK" , 2 : "OK" , 3 : "OK" },
153- "KSTest aa test" : {0 : "OK" , 1 : "OK" , 2 : "OK" , 3 : "OK" },
153+ "KSTest aa test" : {0 : "NOT OK" , 1 : "NOT OK" , 2 : "OK" , 3 : "OK" },
154154 "TTest best split" : {0 : "OK" , 1 : "OK" , 2 : "OK" , 3 : "OK" },
155155 "KSTest best split" : {0 : "OK" , 1 : "OK" , 2 : "OK" , 3 : "OK" },
156156 "result" : {0 : "OK" , 1 : "OK" , 2 : "OK" , 3 : "OK" },
0 commit comments