fixed a few tests

cgfuh · Nov 12, 2013 · 59c5cda · 59c5cda
1 parent d4c62ea
commit 59c5cda
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/polls/models.py b/polls/models.py
@@ -1,5 +1,6 @@
 from django.db import models
 from django.utils import timezone
+import datetime
 
 class Poll(models.Model):
     question = models.CharField(max_length=200)
@@ -18,4 +19,4 @@ class Choice(models.Model):
     votes = models.IntegerField(default=0)
 
     def __unicode__(self):
-        return self.choice_text
+        return self.choice_text
diff --git a/polls/tests.py b/polls/tests.py
@@ -1,9 +1,8 @@
-import datetime
-
-from django.utils import timezone
+from django.core.urlresolvers import reverse
 from django.test import TestCase
-
+from django.utils import timezone
 from polls.models import Poll
+import datetime
 
 def create_poll(question, days):
     """
@@ -94,4 +93,4 @@ def test_index_view_with_two_past_polls(self):
         self.assertQuerysetEqual(
             response.context['latest_poll_list'],
              ['<Poll: Past poll 2.>', '<Poll: Past poll 1.>']
-        )
+        )