at master 1.6 kB view raw
1diff --git a/test/test_datatable_data.py b/test/test_datatable_data.py 2index ee9ac61..0266a19 100644 3--- a/test/test_datatable_data.py 4+++ b/test/test_datatable_data.py 5@@ -1,3 +1,4 @@ 6+import datetime 7 import re 8 import unittest 9 import httpretty 10@@ -135,7 +136,7 @@ class ListDatatableDataTest(unittest.TestCase): 11 df = results.to_pandas() 12 self.assertEqual(df.index.name, 'None') 13 14- # if datatable has Date field then it should be convert to pandas datetime 15+ # if datatable has Date field then it should be convert to datetime 16 @parameterized.expand(['GET', 'POST']) 17 def test_pandas_dataframe_date_field_is_datetime(self, request_method): 18 if request_method == 'POST': 19@@ -143,10 +144,10 @@ class ListDatatableDataTest(unittest.TestCase): 20 datatable = Datatable('ZACKS/FC') 21 results = Data.page(datatable, params={}) 22 df = results.to_pandas() 23- self.assertIsInstance(df['per_end_date'][0], pandas.datetime) 24- self.assertIsInstance(df['per_end_date'][1], pandas.datetime) 25- self.assertIsInstance(df['per_end_date'][2], pandas.datetime) 26- self.assertIsInstance(df['per_end_date'][3], pandas.datetime) 27+ self.assertIsInstance(df['per_end_date'][0], datetime.datetime) 28+ self.assertIsInstance(df['per_end_date'][1], datetime.datetime) 29+ self.assertIsInstance(df['per_end_date'][2], datetime.datetime) 30+ self.assertIsInstance(df['per_end_date'][3], datetime.datetime) 31 32 @parameterized.expand(['GET', 'POST']) 33 def test_to_numpy_returns_numpy_object(self, request_method):