Skip to content

Commit c0742fc

Browse files
committed
Update test_utils.py
1 parent 8c6b555 commit c0742fc

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/acquisition/rvdss/test_utils.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import 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

0 commit comments

Comments
 (0)