diff --git a/lms/djangoapps/grades/management/commands/tests/test_compute_grades.py b/lms/djangoapps/grades/management/commands/tests/test_compute_grades.py index 064c7a2296..0c210ff0f9 100644 --- a/lms/djangoapps/grades/management/commands/tests/test_compute_grades.py +++ b/lms/djangoapps/grades/management/commands/tests/test_compute_grades.py @@ -9,7 +9,7 @@ from __future__ import absolute_import, division, print_function, unicode_litera import ddt from django.contrib.auth import get_user_model from django.core.management import CommandError, call_command -from mock import patch +from mock import ANY, patch import six from student.models import CourseEnrollment @@ -27,14 +27,6 @@ def _sorted_by_batch(calls): return sorted(calls, key=lambda x: (x[1]['kwargs']['course_key'], x[1]['kwargs']['offset'])) -class Any(object): - """ - Dummy object that compares equal to all other objects. - """ - def __eq__(self, other): - return True - - @ddt.ddt class TestComputeGrades(SharedModuleStoreTestCase): """ @@ -109,7 +101,7 @@ class TestComputeGrades(SharedModuleStoreTestCase): 'batch_size': 2, 'offset': offset, 'estimate_first_attempted': estimate_first_attempted, - 'seq_id': Any(), + 'seq_id': ANY, } self.assertEqual( _sorted_by_batch(mock_task.apply_async.call_args_list), @@ -146,7 +138,7 @@ class TestComputeGrades(SharedModuleStoreTestCase): 'batch_size': 2, 'offset': 0, 'estimate_first_attempted': True, - 'seq_id': Any(), + 'seq_id': ANY, }, },), ({ @@ -155,7 +147,7 @@ class TestComputeGrades(SharedModuleStoreTestCase): 'batch_size': 2, 'offset': 2, 'estimate_first_attempted': True, - 'seq_id': Any(), + 'seq_id': ANY, }, },), ],