@@ -148,7 +148,8 @@ def test_expected_groups(self):
148148
149149class TestCheckBadGeoIdFormat :
150150 params = {"data_source" : "" , "span_length" : 0 ,
151- "end_date" : "2020-09-02" , "expected_lag" : {}}
151+ "end_date" : "2020-09-02" , "expected_lag" : {},
152+ "validator_static_file_dir" : "../static" }
152153
153154 def test_empty_df (self ):
154155 validator = Validator (self .params )
@@ -270,7 +271,7 @@ def test_invalid_geo_id_hrr(self):
270271 validator .check_bad_geo_id_value (df , "name" , "hrr" )
271272
272273 assert len (validator .raised_errors ) == 1
273- assert "check_geo_id_value " in validator .raised_errors [0 ].check_data_id
274+ assert "check_bad_geo_id_value " in validator .raised_errors [0 ].check_data_id
274275 assert len (validator .raised_errors [0 ].expression ) == 3
275276 assert "1" not in validator .raised_errors [0 ].expression
276277 assert "11" not in validator .raised_errors [0 ].expression
@@ -285,7 +286,7 @@ def test_invalid_geo_id_state(self):
285286 validator .check_bad_geo_id_value (df , "name" , "state" )
286287
287288 assert len (validator .raised_errors ) == 1
288- assert "check_geo_id_value " in validator .raised_errors [0 ].check_data_id
289+ assert "check_bad_geo_id_value " in validator .raised_errors [0 ].check_data_id
289290 assert len (validator .raised_errors [0 ].expression ) == 1
290291 assert "ak" not in validator .raised_errors [0 ].expression
291292 assert "aa" in validator .raised_errors [0 ].expression
@@ -306,7 +307,7 @@ def test_invalid_geo_id_national(self):
306307 validator .check_bad_geo_id_value (df , "name" , "national" )
307308
308309 assert len (validator .raised_errors ) == 1
309- assert "check_geo_id_value " in validator .raised_errors [0 ].check_data_id
310+ assert "check_bad_geo_id_value " in validator .raised_errors [0 ].check_data_id
310311 assert len (validator .raised_errors [0 ].expression ) == 1
311312 assert "us" not in validator .raised_errors [0 ].expression
312313 assert "zz" in validator .raised_errors [0 ].expression
0 commit comments