improves testcase

somnathrakshit · Sep 21, 2020 · 8566945 · 8566945
1 parent 2997d67
commit 8566945
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_locator.py b/tests/test_locator.py
@@ -86,12 +86,13 @@ def testExamples(self):
         '''
         test examples
         '''
-        examples=['Amsterdam, Netherlands', 'Vienna, Austria','Vienna IL','Paris - Texas', 'Paris TX']
-        countries=['NL','AT','US','US','US']
+        examples=['Amsterdam, Netherlands', 'Vienna, Austria','Vienna IL','Paris - Texas', 'Paris TX',
+                  'Austin, TX','Austin Texas','Auckland, New Zealand']
+        countries=['NL','AT','US','US','US','US','US','NZ']
         for index,example in enumerate(examples):
             city=geograpy.locate(example)
             if self.debug:
-                print(city)
+                print("%22s->%s" % (example,city))
             self.assertEquals(countries[index],city.country.iso)