Merge pull request #104 from ankitjavalkar/view-tests

Add tests for views

1 files changed, 6 insertions, 2 deletions

diff --git a/yaksh/models.py b/yaksh/models.py
index 7035d1e..c48eef1 100644
--- a/yaksh/models.py
+++ b/yaksh/models.py
@@ -42,7 +42,7 @@ test_case_types = (
 
 attempts = [(i, i) for i in range(1, 6)]
 attempts.append((-1, 'Infinite'))
-days_between_attempts = ((j, j) for j in range(401))
+days_between_attempts = [(j, j) for j in range(401)]
 
 test_status = (
                 ('inprogress', 'Inprogress'),
@@ -58,6 +58,10 @@ def get_model_class(model):
 
     return model_class
 
+def has_profile(user):
+    """ check if user has profile """
+    return True if hasattr(user, 'profile') else False
+
 
 ###############################################################################
 class CourseManager(models.Manager):
@@ -159,7 +163,7 @@ class Profile(models.Model):
     department = models.CharField(max_length=64)
     position = models.CharField(max_length=64)
     timezone = models.CharField(max_length=64,
-                                choices=[(tz, tz) for tz in pytz.common_timezones])
+        choices=[(tz, tz) for tz in pytz.common_timezones])
 
 
 ###############################################################################