return None on IndexOutOfRange
This commit is contained in:
parent
b0c5e0e4fd
commit
4887293d2b
|
@ -41,8 +41,14 @@ class Mesure(models.Model):
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def previous(self):
|
def previous(self):
|
||||||
|
try:
|
||||||
return self.gauge.mesures.filter(moment__lt=self.moment).order_by('-moment')[0]
|
return self.gauge.mesures.filter(moment__lt=self.moment).order_by('-moment')[0]
|
||||||
|
except IndexError:
|
||||||
|
return None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def next(self):
|
def next(self):
|
||||||
|
try:
|
||||||
return self.gauge.mesures.filter(moment__gt=self.moment).order_by('moment')[0]
|
return self.gauge.mesures.filter(moment__gt=self.moment).order_by('moment')[0]
|
||||||
|
except IndexError:
|
||||||
|
return None
|
|
@ -74,7 +74,9 @@ class TestMesureModel(TestCase):
|
||||||
|
|
||||||
def test_previous(self):
|
def test_previous(self):
|
||||||
self.assertEqual(self.mesure_1, self.mesure_3.previous)
|
self.assertEqual(self.mesure_1, self.mesure_3.previous)
|
||||||
|
self.assertIsNone(self.mesure_1.previous)
|
||||||
|
|
||||||
def test_next(self):
|
def test_next(self):
|
||||||
self.assertEqual(self.mesure_3, self.mesure_1.next)
|
self.assertEqual(self.mesure_3, self.mesure_1.next)
|
||||||
|
self.assertIsNone(self.mesure_3.next)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue