Bok choy test improvements.

openedx/core/djangoapps/course_groups/tests/test_views.py

@@ -387,8 +387,8 @@ class CohortHandlerTestCase(CohortViewsTestCase):
             self.course, self.cohort1, data={'name': self.cohort1.name, 'group_id': None}
         )
         self.assertEqual((None, None), get_group_info_for_cohort(self.cohort1))
-        self.assertEqual(None, response_dict.get("group_id"))
-        self.assertEqual(None, response_dict.get("user_partition_id"))
+        self.assertIsNone(response_dict.get("group_id"))
+        self.assertIsNone(response_dict.get("user_partition_id"))
 
     def test_change_cohort_group_id(self):
         """

openedx/core/djangoapps/course_groups/views.py

@@ -97,7 +97,6 @@ def cohort_handler(request, course_key_string, cohort_id=None):
                 _get_cohort_representation(c, course)
                 for c in cohorts.get_course_cohorts(course)
             ]
-            # TODO: change to just directly returning the lists.
             return JsonResponse({'cohorts': all_cohorts})
         else:
             cohort = cohorts.get_cohort_by_id(course_key, cohort_id)