Make club app a little more manageable.
[wolnelektury.git] / src / funding / tests.py
index 6dbfbb4..d062809 100644 (file)
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
@@ -27,7 +26,7 @@ class PerksTest(TestCase):
             set())
         self.assertEqual(
             set(self.offer1.get_perks(70)),
-            set([perk, perk1]))
+            {perk, perk1})
 
 
 class FundingTest(TestCase):
@@ -49,11 +48,11 @@ class FundingTest(TestCase):
         self.assertEqual(Offer.current(), self.offer_current)
         self.assertEqual(
             set(Offer.past()),
-            set([self.offer_past])
+            {self.offer_past}
         )
         self.assertEqual(
             set(Offer.public()),
-            set([self.offer_past, self.offer_current])
+            {self.offer_past, self.offer_current}
         )
 
     def test_interrupt(self):
@@ -67,10 +66,9 @@ class FundingTest(TestCase):
         self.assertEqual(Offer.current(), offer_interrupt)
         self.assertEqual(
             set(Offer.past()),
-            set([self.offer_past, self.offer_current])
+            {self.offer_past, self.offer_current}
         )
         self.assertEqual(
             set(Offer.public()),
-            set([self.offer_past, self.offer_current, offer_interrupt])
+            {self.offer_past, self.offer_current, offer_interrupt}
         )
-