Add more test for market_time.py

LiaoU3 · Jul 21, 2024 · 5fc65c1 · 5fc65c1
1 parent 9b90980
commit 5fc65c1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_market_time.py b/tests/test_market_time.py
@@ -23,6 +23,11 @@ def test_init_close_date_set(self):
         )  # not in close_dates
         self.assertTrue(len(mt.close_date_set) == 2)
 
+    def test_init_close_date_set_with_none(self):
+        mt.close_date_set.clear()
+        mt.init_close_date_set(None)
+        self.assertEqual(len(mt.close_date_set), 0)
+
     def test__get_date_obj(self):
         time = datetime.now()
         self.assertTrue(time.date(), mt._get_date_obj(time))  # datetime