Skip to content

Commit e651646

Browse files
committed
add new cols to integration test
1 parent 9db5935 commit e651646

File tree

1 file changed

+55
-2
lines changed

1 file changed

+55
-2
lines changed

integrations/server/test_flusurv.py

Lines changed: 55 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,39 @@ def localSetUp(self):
1111
def test_flusurv(self):
1212
"""Basic integration test for flusurv endpoint"""
1313
self.cur.execute(
14-
"INSERT INTO `flusurv`(`release_date`, `issue`, `epiweek`, `location`, `lag`, `rate_age_0`, `rate_age_1`, `rate_age_2`, `rate_age_3`, `rate_age_4`, `rate_overall`, `rate_age_5`, `rate_age_6`, `rate_age_7`) VALUES(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)",
15-
("2012-11-02", "201243", "201143", "CA", "52", "0", "0", "0", "0.151", "0", "0.029", "0", "0", "0"),
14+
"""
15+
INSERT INTO `flusurv`(
16+
`release_date`, `issue`, `epiweek`, `location`, `lag`,
17+
`rate_age_0`, `rate_age_1`, `rate_age_2`, `rate_age_3`, `rate_age_4`,
18+
`rate_overall`, `rate_age_5`, `rate_age_6`, `rate_age_7`,
19+
`rate_age_18t29`,
20+
`rate_age_30t39`,
21+
`rate_age_40t49`,
22+
`rate_age_5t11`,
23+
`rate_age_12t17`,
24+
`rate_age_lt18`,
25+
`rate_age_gte18`,
26+
`rate_age_1t4`,
27+
`rate_age_gte75`,
28+
`rate_age_0tlt1`,
29+
`rate_race_white`,
30+
`rate_race_black`,
31+
`rate_race_hisp`,
32+
`rate_race_asian`,
33+
`rate_race_natamer`,
34+
`rate_sex_male`,
35+
`rate_sex_female`,
36+
`rate_flu_a`,
37+
`rate_flu_b`,
38+
) VALUES(
39+
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
40+
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s
41+
)
42+
""",
43+
(
44+
"2012-11-02", "201243", "201143", "CA", "52", "0", "0", "0", "0.151", "0", "0.029", "0", "0", "0",
45+
"2.54", "0", "1", "0", "0", "0", "0", "0", "0.68", "0.46", "1", "1", "0", "0", "0", "99", "0", "0", "22.2"
46+
),
1647
)
1748
self.cnx.commit()
1849
response = self.epidata_client.flusurv(epiweeks=201143, locations="CA")
@@ -32,6 +63,28 @@ def test_flusurv(self):
3263
"rate_age_3": 0.151,
3364
"rate_age_4": 0.0,
3465
"rate_overall": 0.029,
66+
"rate_age_5": 0.0,
67+
"rate_age_6": 0.0,
68+
"rate_age_7": 0.0,
69+
"rate_age_18t29": 2.54,
70+
"rate_age_30t39": 0.0,
71+
"rate_age_40t49": 1.0,
72+
"rate_age_5t11": 0.0,
73+
"rate_age_12t17": 0.0,
74+
"rate_age_lt18": 0.0,
75+
"rate_age_gte18": 0.0,
76+
"rate_age_1t4": 0.0,
77+
"rate_age_gte75": 0.68,
78+
"rate_age_0tlt1": 0.46,
79+
"rate_race_white": 1.0,
80+
"rate_race_black": 1.0,
81+
"rate_race_hisp": 0.0,
82+
"rate_race_asian": 0.0,
83+
"rate_race_natamer": 0.0,
84+
"rate_sex_male": 99.0,
85+
"rate_sex_female": 0.0,
86+
"rate_flu_a": 0.0,
87+
"rate_flu_b": 22.2,
3588
}
3689
],
3790
"result": 1,

0 commit comments

Comments
 (0)