def test_population_by_country_for_non_empty_course(self):
course = CourseFactory.create()
empty_course = CourseFactory.create()
course_id = self.get_course_id(course)
empty_course_id = self.get_course_id(empty_course)
CourseEnrollmentFactory.create(course_id=course_id, user__profile__country='FR')
empty_course_population = stats.population_by_country(empty_course_id)
course_population = stats.population_by_country(course_id)
self.assertEqual({}, empty_course_population)
self.assertEqual({'FR': 1}, course_population)
def test_population_by_country_for_non_empty_course(self):
course = CourseFactory.create()
empty_course = CourseFactory.create()
course_id = self.get_course_id(course)
empty_course_id = self.get_course_id(empty_course)
CourseEnrollmentFactory.create(course_id=course_id,
user__profile__country='FR')
empty_course_population = stats.population_by_country(empty_course_id)
course_population = stats.population_by_country(course_id)
self.assertEqual({}, empty_course_population)
self.assertEqual({'FR': 1}, course_population)
def test_inactive_enrollments_are_not_included(self):
course = CourseFactory.create()
self.enroll_student(course,
user__profile__country='FR',
user__is_active=False)
course_population = stats.population_by_country(
self.get_course_id(course))
self.assertEqual({'FR': 1}, course_population)
def test_null_countries_are_counted(self):
course = CourseFactory.create()
course_id = self.get_course_id(course)
CourseEnrollmentFactory.create(course_id=course_id, user__profile__country=None)
CourseEnrollmentFactory.create(course_id=course_id, user__profile__country=None)
course_population = stats.population_by_country(course_id)
self.assertEqual(1, len(course_population))
self.assertEqual(2, course_population[countries.UNKNOWN_COUNTRY_CODE])
def test_dependent_territories_are_not_listed_separately(self):
course = CourseFactory.create()
self.enroll_student(course,
user__profile__country='GF',
user__is_active=False)
self.enroll_student(course,
user__profile__country='PF',
user__is_active=False)
course_population = stats.population_by_country(
self.get_course_id(course))
self.assertEqual({'FR': 2}, course_population)
def test_null_and_empty_countries_are_grouped(self):
course = CourseFactory.create()
course_id = self.get_course_id(course)
CourseEnrollmentFactory.create(course_id=course_id,
user__profile__country=None)
CourseEnrollmentFactory.create(course_id=course_id,
user__profile__country='')
course_population = stats.population_by_country(course_id)
self.assertEqual(1, len(course_population))
self.assertEqual(2, course_population[countries.UNKNOWN_COUNTRY_CODE])
def test_inactive_enrollments_are_not_included(self):
course = CourseFactory.create()
self.enroll_student(course, user__profile__country='FR', user__is_active=False)
course_population = stats.population_by_country(self.get_course_id(course))
self.assertEqual({'FR': 1}, course_population)
def test_dependent_territories_are_not_listed_separately(self):
course = CourseFactory.create()
self.enroll_student(course, user__profile__country='GF', user__is_active=False)
self.enroll_student(course, user__profile__country='PF', user__is_active=False)
course_population = stats.population_by_country(self.get_course_id(course))
self.assertEqual({'FR': 2}, course_population)
def test_population_by_country_for_empty_course(self):
course = CourseFactory.create()
course_population = stats.population_by_country(self.get_course_id(course))
self.assertEqual({}, course_population)
def test_population_by_country_for_empty_course(self):
course = CourseFactory.create()
course_population = stats.population_by_country(
self.get_course_id(course))
self.assertEqual({}, course_population)
