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 2
2
3
3
import pytest
4
4
5
- from delphi .epidata .acquisition .rvdss .utils import abbreviate_virus
5
+ from delphi .epidata .acquisition .rvdss .utils import abbreviate_virus , create_geo_types
6
6
7
7
# py3tester coverage target
8
8
__test_target__ = "delphi.epidata.acquisition.rvdss.utils"
@@ -16,3 +16,9 @@ def test_syntax(self):
16
16
def test_abbreviate_virus (self ):
17
17
assert abbreviate_virus ("influenza" ) == "flu" # normal case
18
18
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