BUG: Fix missing import for datatime_data.

2 files changed, 8 insertions, 0 deletions

diff --git a/numpy/lib/tests/test_type_check.py b/numpy/lib/tests/test_type_check.py
index 01494594b..751392fef 100644
--- a/numpy/lib/tests/test_type_check.py
+++ b/numpy/lib/tests/test_type_check.py
@@ -375,5 +375,12 @@ class TestArrayConversion(TestCase):
         assert issubdtype(a.dtype,float)
 
 
+class TestDateTimeData:
+
+    def test_basic(self):
+        a = array(['1980-03-23'], dtype=datetime64)
+        assert_equal(datetime_data(a.dtype), ('us', 1, 1, 1))
+
+
 if __name__ == "__main__":
     run_module_suite()
diff --git a/numpy/lib/type_check.py b/numpy/lib/type_check.py
index 7c88b192d..05070b3ba 100644
--- a/numpy/lib/type_check.py
+++ b/numpy/lib/type_check.py
@@ -619,6 +619,7 @@ def datetime_data(dtype):
                     ('den', ctypes.c_int),
                     ('events', ctypes.c_int)]
 
+    import sys
     if sys.version_info[:2] >= (3,1):
         func = ctypes.pythonapi.PyCapsule_GetPointer
         func.argtypes = [ctypes.py_object, ctypes.c_char_p]