def test_get_country(self):
# Countries should be accessible 2 or 3 digit country code
self.assertEqual(get_country('US'), get_country('USA'))
# Use common name for Taiwan
self.assertEqual(get_country('TW').name, 'Taiwan')
# Return unknown country if code is invalid
self.assertEqual(get_country('A1'), UNKNOWN_COUNTRY)
self.assertEqual(get_country(None), UNKNOWN_COUNTRY)
def test_get_country(self):
# Countries should be accessible 2 or 3 digit country code
self.assertEqual(get_country('US'), get_country('USA'))
# Use common name for Taiwan
self.assertEqual(get_country('TW').name, 'Taiwan')
# Return unknown country if code is invalid
self.assertEqual(get_country('A1'), UNKNOWN_COUNTRY)
self.assertEqual(get_country(None), UNKNOWN_COUNTRY)
def country(self):
"""
Returns a Country object representing the country in this model's country_code.
"""
return country.get_country(self.country_code)
def test_country(self):
country = get_country('US')
self.assertEqual(country.alpha2, 'US')
instance = G(models.CourseEnrollmentByCountry,
country_code=country.alpha2)
self.assertEqual(instance.country, country)
def setUp(self):
super(CourseEnrollmentByLocationViewTests, self).setUp()
self.country = get_country('US')
self.generate_data()
def setUpClass(cls):
super().setUpClass()
cls.country = get_country('US')
def country(self):
"""
Returns a Country object representing the country in this model's country_code.
"""
return country.get_country(self.country_code)
def test_country(self):
country = get_country('US')
self.assertEqual(country.alpha2, 'US')
instance = G(models.CourseEnrollmentByCountry, country_code=country.alpha2)
self.assertEqual(instance.country, country)
def setUp(self):
super(CourseEnrollmentByLocationViewTests, self).setUp()
self.country = get_country('US')
self.generate_data()
def setUpClass(cls):
super(CourseEnrollmentByLocationViewTests, cls).setUpClass()
cls.country = get_country('US')
def setUpClass(cls):
super(CourseEnrollmentByLocationViewTests, cls).setUpClass()
cls.country = get_country('US')
