Merge pull request #115 from fuzzyhandle/master

Special situations where weekend is a working day

nsepy/live.py

@@ -202,5 +202,23 @@ def getworkingdays(dtfrom, dtto):
             stweekends.add(dt)
 
     # pdb.set_trace()
+    stspecial  = set(
+      [datetime.date(2020,2,1) # Budget day
+      ]
+    )
+
+    #Remove special weekend working days from weekends set
+    stweekends -= stspecial
     stworking = (stalldays - stweekends) - set(dfholiday.index.values)
-    return sorted(stworking)
+    # stworking = (stalldays - stweekends) - set(dfholiday.index.values)
+
+    # #Special cases where market was open on weekends
+    # stspecial  = set(
+      # [datetime.date(2020,2,1) # Budget day
+      # ]
+    # )
+    # for dtspecial in stspecial:
+      # if (dtspecial >= dtfrom and dtspecial <= dtto):
+        # stworking.add(dtspecial)
+
+    return sorted(stworking)

tests/test_live.py

@@ -150,3 +150,9 @@ def test_working_day(self):
         workingdays = getworkingdays(datetime.date(
             2019, 3, 1), datetime.date(2019, 3, 31))
         self.assertEqual(len(workingdays), 19)
+
+        # working day for special dates on weekend
+        workingdays = getworkingdays(datetime.date(
+            2020, 1, 31), datetime.date(2020, 2, 3))
+        self.assertEqual(len(workingdays), 3)
+