From 8ac0150914c4a47292b788435c316815f2757b05 Mon Sep 17 00:00:00 2001 From: Fred Date: Fri, 5 Aug 2016 21:32:08 +0200 Subject: [PATCH] define setUp in TestMesureModel --- src/sherlock/tests.py | 24 +++++++++++------------- 1 file changed, 11 insertions(+), 13 deletions(-) diff --git a/src/sherlock/tests.py b/src/sherlock/tests.py index b7a602e..80f09e4 100644 --- a/src/sherlock/tests.py +++ b/src/sherlock/tests.py @@ -38,22 +38,20 @@ class TestGaugeModel(TestCase): class TestMesureModel(TestCase): - def test_category_property(self): - cat = Category.objects.create(name='eau') - gauge = Gauge.objects.create(serial_number='undefined', category=cat) - mesure = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.001, gauge=gauge) + def setUp(self): + self.cat_water = Category.objects.create(name='eau') + self.cat_gaz = Category.objects.create(name='gaz') - self.assertEqual(cat, mesure.category) + self.gauge_1 = Gauge.objects.create(serial_number='undefined', category=self.cat_water) + self.gauge_2 = Gauge.objects.create(serial_number='undefined', category=self.cat_gaz) + + self.mesure_1 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.001, gauge=self.gauge_1) + self.mesure_2 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.002, gauge=self.gauge_2) + + def test_category_property(self): + self.assertEqual(self.cat_water, self.mesure_1.category) def test_diff_raises_exception(self): - cat_water = Category.objects.create(name='eau') - cat_gaz = Category.objects.create(name='gaz') - - gauge_1 = Gauge.objects.create(serial_number='undefined', category=cat_water) - gauge_2 = Gauge.objects.create(serial_number='undefined', category=cat_gaz) - - mesure_1 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.001, gauge=gauge_1) - mesure_2 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.002, gauge=gauge_2) with self.assertRaises(Exception): mesure_1.diff(mesure_2)