From 4453c80634fb72b60e849480e52c06b570c696d5 Mon Sep 17 00:00:00 2001 From: Florian Apolloner Date: Wed, 30 Jul 2014 22:22:04 +0200 Subject: [PATCH] Fixed the previous commit for Python3. --- tests/model_fields/models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/model_fields/models.py b/tests/model_fields/models.py index 8dfdcb91d5..0e9fe23c67 100644 --- a/tests/model_fields/models.py +++ b/tests/model_fields/models.py @@ -10,6 +10,7 @@ except ImportError: from django.core.files.storage import FileSystemStorage from django.db import models from django.db.models.fields.files import ImageFieldFile, ImageField +from django.utils import six class Foo(models.Model): @@ -43,14 +44,14 @@ class Whiz(models.Model): c = models.IntegerField(choices=CHOICES, null=True) -class Counter: +class Counter(six.Iterator): def __init__(self): self.n = 1 def __iter__(self): return self - def next(self): # Python 3: def __next__(self) + def __next__(self): if self.n > 5: raise StopIteration else: