34 lines
1.6 KiB
Diff
34 lines
1.6 KiB
Diff
|
diff --git a/test/test_datatable_data.py b/test/test_datatable_data.py
|
||
|
|
index ee9ac61..0266a19 100644
|
||
|
|
--- a/test/test_datatable_data.py
|
||
|
|
+++ b/test/test_datatable_data.py
|
||
|
|
@@ -1,3 +1,4 @@
|
||
|
|
+import datetime
|
||
|
|
import re
|
||
|
|
import unittest
|
||
|
|
import httpretty
|
||
|
|
@@ -135,7 +136,7 @@ class ListDatatableDataTest(unittest.TestCase):
|
||
|
|
df = results.to_pandas()
|
||
|
|
self.assertEqual(df.index.name, 'None')
|
||
|
|
|
||
|
|
- # if datatable has Date field then it should be convert to pandas datetime
|
||
|
|
+ # if datatable has Date field then it should be convert to datetime
|
||
|
|
@parameterized.expand(['GET', 'POST'])
|
||
|
|
def test_pandas_dataframe_date_field_is_datetime(self, request_method):
|
||
|
|
if request_method == 'POST':
|
||
|
|
@@ -143,10 +144,10 @@ class ListDatatableDataTest(unittest.TestCase):
|
||
|
|
datatable = Datatable('ZACKS/FC')
|
||
|
|
results = Data.page(datatable, params={})
|
||
|
|
df = results.to_pandas()
|
||
|
|
- self.assertIsInstance(df['per_end_date'][0], pandas.datetime)
|
||
|
|
- self.assertIsInstance(df['per_end_date'][1], pandas.datetime)
|
||
|
|
- self.assertIsInstance(df['per_end_date'][2], pandas.datetime)
|
||
|
|
- self.assertIsInstance(df['per_end_date'][3], pandas.datetime)
|
||
|
|
+ self.assertIsInstance(df['per_end_date'][0], datetime.datetime)
|
||
|
|
+ self.assertIsInstance(df['per_end_date'][1], datetime.datetime)
|
||
|
|
+ self.assertIsInstance(df['per_end_date'][2], datetime.datetime)
|
||
|
|
+ self.assertIsInstance(df['per_end_date'][3], datetime.datetime)
|
||
|
|
|
||
|
|
@parameterized.expand(['GET', 'POST'])
|
||
|
|
def test_to_numpy_returns_numpy_object(self, request_method):
|