1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
|
from unittest import TestCase
from ..usa import (
UnitedStates,
Alabama,
Alaska,
Arizona,
Arkansas,
California,
Colorado,
Connecticut,
Delaware,
DistrictOfColumbia,
Florida,
Georgia,
Hawaii,
Idaho,
Illinois,
Indiana,
Iowa,
Kansas,
Kentucky,
Louisiana,
Maine,
Maryland,
Massachusetts,
Michigan,
Minnesota,
Mississippi,
Missouri,
Montana,
Nebraska,
Nevada,
NewHampshire,
NewJersey,
NewMexico,
NewYork,
NorthCarolina,
NorthDakota,
Ohio,
Oklahoma,
Oregon,
Pennsylvania,
RhodeIsland,
SouthCarolina,
SouthDakota,
Tennessee,
Texas,
Utah,
Vermont,
Virginia,
Washington,
WestVirginia,
Wisconsin,
Wyoming
)
from ..registry import registry
class RegistryUsa(TestCase):
def _check_all_states(self, classes):
self.assertIn(Alabama, classes)
self.assertIn(Alaska, classes)
self.assertIn(Arizona, classes)
self.assertIn(Arkansas, classes)
self.assertIn(California, classes)
self.assertIn(Colorado, classes)
self.assertIn(Connecticut, classes)
self.assertIn(Delaware, classes)
self.assertIn(DistrictOfColumbia, classes)
self.assertIn(Florida, classes)
self.assertIn(Georgia, classes)
self.assertIn(Hawaii, classes)
self.assertIn(Idaho, classes)
self.assertIn(Illinois, classes)
self.assertIn(Indiana, classes)
self.assertIn(Iowa, classes)
self.assertIn(Kansas, classes)
self.assertIn(Kentucky, classes)
self.assertIn(Louisiana, classes)
self.assertIn(Maine, classes)
self.assertIn(Maryland, classes)
self.assertIn(Massachusetts, classes)
self.assertIn(Michigan, classes)
self.assertIn(Minnesota, classes)
self.assertIn(Mississippi, classes)
self.assertIn(Missouri, classes)
self.assertIn(Montana, classes)
self.assertIn(Nebraska, classes)
self.assertIn(Nevada, classes)
self.assertIn(NewHampshire, classes)
self.assertIn(NewJersey, classes)
self.assertIn(NewMexico, classes)
self.assertIn(NewYork, classes)
self.assertIn(NorthCarolina, classes)
self.assertIn(NorthDakota, classes)
self.assertIn(Ohio, classes)
self.assertIn(Oklahoma, classes)
self.assertIn(Oregon, classes)
self.assertIn(Pennsylvania, classes)
self.assertIn(RhodeIsland, classes)
self.assertIn(SouthCarolina, classes)
self.assertIn(SouthDakota, classes)
self.assertIn(Tennessee, classes)
self.assertIn(Texas, classes)
self.assertIn(Utah, classes)
self.assertIn(Vermont, classes)
self.assertIn(Virginia, classes)
self.assertIn(Washington, classes)
self.assertIn(WestVirginia, classes)
self.assertIn(Wisconsin, classes)
self.assertIn(Wyoming, classes)
def test_usa_world(self):
classes = set(registry.region_registry.values())
self._check_all_states(classes)
# On top of it, the core class
self.assertIn(UnitedStates, classes)
def test_usa_subregion(self):
# Get all the subregions
classes = set(registry.get_subregions('US').values())
self._check_all_states(classes)
|