File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change 22
33import pytest
44
5- from delphi .epidata .acquisition .rvdss .utils import abbreviate_virus
5+ from delphi .epidata .acquisition .rvdss .utils import abbreviate_virus , create_geo_types
66
77# py3tester coverage target
88__test_target__ = "delphi.epidata.acquisition.rvdss.utils"
@@ -16,3 +16,9 @@ def test_syntax(self):
1616 def test_abbreviate_virus (self ):
1717 assert abbreviate_virus ("influenza" ) == "flu" # normal case
1818 assert abbreviate_virus ("flu" ) == "flu" # already abbreviated
19+
20+ def test_create_geo_types (self ):
21+ assert create_geo_types ("canada" ,"lab" ) == "nation"
22+ assert create_geo_types ("bc" ,"lab" ) == "region"
23+ assert create_geo_types ("random lab" ,"lab" ) == "lab"
24+ assert create_geo_types ("Canada" ,"province" ) == "province" #lowercase handling happens upstream
You can’t perform that action at this time.
0 commit comments