'Test get_cohorted_user_partition returns first user_partition with scheme `CohortPartitionScheme`.'

def test_returns_first_cohort_user_partition(self):

    self.course.user_partitions.append(self.random_user_partition)
    self.course.user_partitions.append(self.cohort_user_partition)
    self.course.user_partitions.append(self.second_cohort_user_partition)
    self.assertEqual(self.cohort_user_partition, get_cohorted_user_partition(self.course))
