TST: Parametrize test_wom_len

Author: asdf8601

pandas/tests/indexes/datetimes/test_date_range.py

@@ -236,10 +236,12 @@ def test_catch_infinite_loop(self):
         pytest.raises(Exception, date_range, datetime(2011, 11, 11),
                       datetime(2011, 11, 12), freq=offset)
 
-    def test_wom_len_one(self):
+    @pytest.mark.parametrize('periods', (1, 2))
+    def test_wom_len(self, periods):
         # https://github.com/pandas-dev/pandas/issues/20517
-        res = date_range(start='20110101', periods=1, freq='WOM-1MON')
-        assert len(res) == 1
+        res = date_range(start='20110101', periods=periods, freq='WOM-1MON')
+        assert len(res) == periods
+
 
 class TestGenRangeGeneration(object):