@@ -92,11 +92,13 @@ def test__geographical_pooling(self, min_obs, expected):
92
92
@pytest .mark .parametrize ("min_obs, expected_pos_prop, expected_se, expected_sample_sz" , [
93
93
(3 , # one case of tests < min_obs
94
94
np .array ([np .nan , 2.5 / 5 , 3.5 / 7 , 4.5 / 11 ]),
95
- np .array ([np .nan , np .sqrt (2.5 ** 2 / 5 ** 2 / 4 ), np .sqrt (3.5 ** 2 / 7 ** 2 / 6 ), np .sqrt (4.5 * 6.5 / 11 ** 2 / 10 )]),
95
+ np .array ([np .nan , np .sqrt (2.5 * (5 - 2.5 )/ 5 / 5 / 4 ), np .sqrt (3.5 * (7 - 3.5 )/ 7 / 7 / 6 ),
96
+ np .sqrt (4.5 * (11 - 4.5 )/ 11 / 11 / 10 )]),
96
97
np .array ([np .nan , 4 , 6 , 10 ])),
97
98
(1 , # no cases of tests < min_obs
98
99
np .array ([1.5 / 3 , 2.5 / 5 , 3.5 / 7 , 4.5 / 11 ]),
99
- np .array ([np .sqrt (1.5 ** 2 / 3 ** 2 / 2 ), np .sqrt (2.5 ** 2 / 5 ** 2 / 4 ), np .sqrt (3.5 ** 2 / 7 ** 2 / 6 ), np .sqrt (4.5 * 6.5 / 11 ** 2 / 10 )]),
100
+ np .array ([np .sqrt (1.5 * (3 - 1.5 )/ 3 / 3 / 2 ), np .sqrt (2.5 * (5 - 2.5 )/ 5 / 5 / 4 ),
101
+ np .sqrt (3.5 * (7 - 3.5 )/ 7 / 7 / 6 ), np .sqrt (4.5 * (11 - 4.5 )/ 11 / 11 / 10 )]),
100
102
np .array ([2 , 4 , 6 , 10 ])),
101
103
])
102
104
def test_raw_positive_prop (self , min_obs , expected_pos_prop , expected_se , expected_sample_sz ):
@@ -124,15 +126,17 @@ def test_raw_positive_prop(self, min_obs, expected_pos_prop, expected_se, expect
124
126
None ,
125
127
None ,
126
128
np .array ([np .nan , 3.5 / 7 , 5.5 / 11 , 7.5 / 17 ]),
127
- np .array ([np .nan , np .sqrt (3.5 ** 2 / 7 ** 2 / 6 ), np .sqrt (5.5 ** 2 / 11 ** 2 / 10 ), np .sqrt (7.5 * 9.5 / 17 ** 2 / 16 )]),
129
+ np .array ([np .nan , np .sqrt (3.5 * (7 - 3.5 )/ 7 / 7 / 6 ), np .sqrt (5.5 * (11 - 5.5 )/ 11 / 11 / 10 ),
130
+ np .sqrt (7.5 * (17 - 7.5 )/ 17 / 17 / 16 )]),
128
131
np .array ([np .nan , 6 , 10 , 16 ]),
129
132
),
130
133
(3 , # parents case
131
134
2 ,
132
135
np .array ([3 , 7 , 9 , 11 ]),
133
136
np .array ([5 , 10 , 15 , 20 ]),
134
137
np .array ([(1 + 0.6 + 0.5 )/ (2 + 1 + 1 ), 3.5 / 7 , 5.5 / 11 , 7.5 / 17 ]),
135
- np .array ([np .sqrt (2.1 * 1.9 / 4 ** 2 / 3 ), np .sqrt (3.5 ** 2 / 7 ** 2 / 6 ), np .sqrt (5.5 ** 2 / 11 ** 2 / 10 ), np .sqrt (7.5 * 9.5 / 17 ** 2 / 16 )]),
138
+ np .array ([np .sqrt (2.1 * (4 - 2.1 )/ 4 / 4 / 3 ), np .sqrt (3.5 * (7 - 3.5 )/ 7 / 7 / 6 ),
139
+ np .sqrt (5.5 * (11 - 5.5 )/ 11 / 11 / 10 ), np .sqrt (7.5 * (17 - 7.5 )/ 17 / 17 / 16 )]),
136
140
np .array ([3 , 6 , 10 , 16 ]),
137
141
),
138
142
])
0 commit comments