Refactored unit test.

master
Brandon Taylor 2015-07-05 23:02:44 -04:00
parent bfcd11f072
commit 51ed60a127
1 changed files with 6 additions and 9 deletions

View File

@ -178,19 +178,16 @@ class SortableTestCase(TestCase):
self.people[next_index])) self.people[next_index]))
def test_get_previous(self): def test_get_previous(self):
for person in self.people.order_by('-order'): for person in self.people:
previous_person = person.get_previous() previous_person = person.get_previous()
# get_previous() returns `None` if there isn't a previous object # get_previous() returns `None` if there isn't a previous object
if previous_person: if previous_person:
previous_order = previous_person.order - 1 self.assertEqual(previous_person,
self.people.get(order=person.order - 1),
# only check for a matching previous record if the order is 'Previous person for "{0}" (order: {1}) should be "{2}"'
# greater than 0 '(order: {3})'.format(person, person.order,
if previous_order > 0: previous_person, previous_person.order))
self.assertEqual(previous_person,
self.people.get(order=previous_person.order - 1),
'Previous person should be "{0}"'.format(previous_person))
def test_adminsortable_change_list_view_loads_with_sortable_fk(self): def test_adminsortable_change_list_view_loads_with_sortable_fk(self):
category1 = self.create_category(title='Category 3') category1 = self.create_category(title='Category 3')