From 3bf4ca9c61cb4f31e0ccbb9a2c86d010605d2bd7 Mon Sep 17 00:00:00 2001
From: gouravkr <gourav2711@gmail.com>
Date: Thu, 24 Feb 2022 08:16:45 +0530
Subject: [PATCH] Added __contains__ and some tests for it

---
 fincal/core.py     |  4 ++++
 tests/test_core.py | 13 ++++++++++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/fincal/core.py b/fincal/core.py
index 94f3bcc..de53cb5 100644
--- a/fincal/core.py
+++ b/fincal/core.py
@@ -387,6 +387,10 @@ class TimeSeriesCore(UserDict):
             self.n += 1
             return key, self.data[key]
 
+    def __contains__(self, key: object) -> bool:
+        key = _parse_date(key)
+        return super().__contains__(key)
+
     def head(self, n: int = 6):
         """Returns the first n items of the TimeSeries object"""
 
diff --git a/tests/test_core.py b/tests/test_core.py
index 8a64acb..343c673 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -3,7 +3,6 @@ from typing import Mapping
 
 from fincal.core import AllFrequencies, Frequency, Series, TimeSeriesCore
 from fincal.fincal import create_date_series
-from numpy import isin
 
 
 class TestFrequency:
@@ -68,3 +67,15 @@ class TestTimeSeriesCore:
         assert ts[ts.dates == '2021-02-01'].iloc[0][1] == 230
         assert ts.iloc[2][0] == datetime.datetime(2021, 3, 1)
         assert len(ts.iloc[:2]) == 2
+
+    def test_contains(self):
+        ts = TimeSeriesCore(self.data, frequency='M')
+        assert datetime.datetime(2021, 1, 1) in ts
+        assert '2021-01-01' in ts
+        assert '2021-01-14' not in ts
+
+    def test_items(self):
+        ts = TimeSeriesCore(self.data, frequency='M')
+        for i, j in ts.items():
+            assert j == self.data[0][1]
+            break