Repository: joke2k/faker
Pull request #523 (2017-05-30 14:47:32)

Make static data accessible at class level in pl_PL DateTime provider
This cleans tests by giving direct access to dataset used to generate results


Patch:
diff --git a/faker/providers/date_time/pl_PL/__init__.py b/faker/providers/date_time/pl_PL/__init__.py
index 5d71a5c4..595c3ce3 100644
--- a/faker/providers/date_time/pl_PL/__init__.py
+++ b/faker/providers/date_time/pl_PL/__init__.py
@@ -7,35 +7,37 @@ from .. import Provider as DateTimeProvider
 
 class Provider(DateTimeProvider):
 
+    DAY_NAMES = {
+        '0': 'poniedziałek',
+        '1': 'wtorek',
+        '2': 'środa',
+        '3': 'czwartek',
+        '4': 'piątek',
+        '5': 'sobota',
+        '6': 'niedziela',
+    }
+
+    MONTH_NAMES = {
+        '01': 'styczeń',
+        '02': 'luty',
+        '03': 'marzec',
+        '04': 'kwiecień',
+        '05': 'maj',
+        '06': 'czerwiec',
+        '07': 'lipiec',
+        '08': 'sierpień',
+        '09': 'wrzesień',
+        '10': 'październik',
+        '11': 'listopad',
+        '12': 'grudzień'
+    }
+
     @classmethod
     def day_of_week(cls):
         day = cls.date('%w')
-        DAY_NAMES = {
-            '0': 'poniedziałek',
-            '1': 'wtorek',
-            '2': 'środa',
-            '3': 'czwartek',
-            '4': 'piątek',
-            '5': 'sobota',
-            '6': 'niedziela',
-        }
-        return DAY_NAMES[day]
+        return cls.DAY_NAMES[day]
 
     @classmethod
     def month_name(cls):
         month = cls.month()
-        MONTH_NAMES = {
-            '01': 'styczeń',
-            '02': 'luty',
-            '03': 'marzec',
-            '04': 'kwiecień',
-            '05': 'maj',
-            '06': 'czerwiec',
-            '07': 'lipiec',
-            '08': 'sierpień',
-            '09': 'wrzesień',
-            '10': 'październik',
-            '11': 'listopad',
-            '12': 'grudzień'
-        }
-        return MONTH_NAMES[month]
+        return cls.MONTH_NAMES[month]


Test patch:
diff --git a/tests/providers/date_time.py b/tests/providers/date_time.py
index c4910c2d..20079207 100644
--- a/tests/providers/date_time.py
+++ b/tests/providers/date_time.py
@@ -24,38 +24,14 @@ class TestHuHU(unittest.TestCase):
 
 class TestPlPL(unittest.TestCase):
 
-    DAY_NAMES = (
-        'poniedziałek',
-        'wtorek',
-        'środa',
-        'czwartek',
-        'piątek',
-        'sobota',
-        'niedziela',
-    )
-
-    MONTH_NAMES = (
-        'styczeń',
-        'luty',
-        'marzec',
-        'kwiecień',
-        'maj',
-        'czerwiec',
-        'lipiec',
-        'sierpień',
-        'wrzesień',
-        'październik',
-        'listopad',
-        'grudzień'
-    )
-
     def setUp(self):
         self.factory = Factory.create('pl_PL')
+        self.provider = self.factory.provider('faker.providers.date_time')
 
     def test_day(self):
         day = self.factory.day_of_week()
-        assert day in self.DAY_NAMES
+        assert day in self.provider.DAY_NAMES.values()
 
     def test_month(self):
         month = self.factory.month_name()
-        assert month in self.MONTH_NAMES
+        assert month in self.provider.MONTH_NAMES.values()