From 15b0b96488f8cd77b8685ba7cb5eefb652221015 Mon Sep 17 00:00:00 2001
From: Gourav Kumar <gourav2711@gmail.com>
Date: Tue, 22 Feb 2022 11:29:07 +0530
Subject: [PATCH] Moved frequency test from test_fincal Added tests for Series
 class

---
 tests/test_core.py | 49 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 tests/test_core.py

diff --git a/tests/test_core.py b/tests/test_core.py
new file mode 100644
index 0000000..65d754c
--- /dev/null
+++ b/tests/test_core.py
@@ -0,0 +1,49 @@
+import datetime
+
+from fincal.core import AllFrequencies, Frequency, Series
+from fincal.fincal import create_date_series
+
+
+class TestFrequency:
+    def test_creation(self):
+        D = Frequency('daily', 'days', 1, 1, 'D')
+        assert D.days == 1
+        assert D.symbol == 'D'
+        assert D.name == 'daily'
+        assert D.value == 1
+        assert D.freq_type == 'days'
+
+
+class TestAllFrequencies:
+    def test_attributes(self):
+        assert hasattr(AllFrequencies, 'D')
+        assert hasattr(AllFrequencies, 'M')
+        assert hasattr(AllFrequencies, 'Q')
+
+    def test_days(self):
+        assert AllFrequencies.D.days == 1
+        assert AllFrequencies.M.days == 30
+        assert AllFrequencies.Q.days == 91
+
+    def test_symbol(self):
+        assert AllFrequencies.H.symbol == 'H'
+        assert AllFrequencies.W.symbol == 'W'
+
+    def test_values(self):
+        assert AllFrequencies.H.value == 6
+        assert AllFrequencies.Y.value == 1
+
+    def test_type(self):
+        assert AllFrequencies.Q.freq_type == 'months'
+        assert AllFrequencies.W.freq_type == 'days'
+
+
+class TestSeries:
+    def test_creation(self):
+        series = Series([1, 2, 3, 4, 5, 6, 7], data_type=int)
+        assert series.dtype == float
+        assert series[2] == 3
+
+        dates = create_date_series('2021-01-01', '2021-01-31', 'D')
+        series = Series(dates, data_type=datetime.datetime)
+        assert Series.dtype == datetime.datetime