add support for datetime64 and timedelta64

Author: Christopher Rowley

docs/src/juliacall-reference.md

@@ -203,7 +203,7 @@ jl.Vector[jl.Int]()
 
 Some Julia types can be converted to corresponding numpy dtypes like `numpy.dtype(jl.Int)`.
 Currently supports these primitive types: `Bool`, `IntXX`, `UIntXX`, `FloatXX`,
-`ComplexFXX`, `Ptr{Cvoid}`.
+`ComplexFXX`, `NumpyDates.InlineDateTime64{unit}` and `NumpyDates.InlineTimeDelta64{unit}`.
 `````
 
 `````@customdoc

src/JlWrap/type.jl

@@ -12,40 +12,17 @@ function pyjltype_getitem(self::Type, k_)
 end
 
 function pyjltype_numpy_dtype(self::Type)
+    typestr, descr = pytypestrdescr(self)
+    if isempty(typestr)
+        errset(pybuiltins.AttributeError, "__numpy_dtype__")
+        return PyNULL
+    end
     np = pyimport("numpy")
-    if self === Bool
-        return np.dtype(np.bool_)
-    elseif self === Int8
-        return np.dtype(np.int8)
-    elseif self === Int16
-        return np.dtype(np.int16)
-    elseif self === Int32
-        return np.dtype(np.int32)
-    elseif self === Int64
-        return np.dtype(np.int64)
-    elseif self === UInt8
-        return np.dtype(np.uint8)
-    elseif self === UInt16
-        return np.dtype(np.uint16)
-    elseif self === UInt32
-        return np.dtype(np.uint32)
-    elseif self === UInt64
-        return np.dtype(np.uint64)
-    elseif self === Float16
-        return np.dtype(np.float16)
-    elseif self === Float32
-        return np.dtype(np.float32)
-    elseif self === Float64
-        return np.dtype(np.float64)
-    elseif self === ComplexF32
-        return np.dtype(np.complex64)
-    elseif self === ComplexF64
-        return np.dtype(np.complex128)
-    elseif self === Ptr{Cvoid}
-        return np.dtype("P")
+    if pyisnull(descr)
+        return np.dtype(typestr)
+    else
+        return np.dtype(descr)
     end
-    errset(pybuiltins.AttributeError, "__numpy_dtype__")
-    return PyNULL
 end
 
 pyjl_handle_error_type(::typeof(pyjltype_numpy_dtype), x, exc) = pybuiltins.AttributeError

test/JlWrap.jl

@@ -473,6 +473,7 @@ end
 end
 
 @testitem "type" setup = [Setup] begin
+    using PythonCall.NumpyDates
     @testset "type" begin
         @test pyis(pytype(pyjl(Int)), PythonCall.pyjltypetype)
     end
@@ -499,17 +500,22 @@ end
                 (Float64, "float64"),
                 (ComplexF32, "complex64"),
                 (ComplexF64, "complex128"),
-                (Ptr{Cvoid}, "P"),
+                (InlineDateTime64{SECONDS}, "datetime64[s]"),
+                (InlineDateTime64{(SECONDS, 5)}, "datetime64[5s]"),
+                (InlineDateTime64{NumpyDates.UNBOUND_UNITS}, "datetime64"),
+                (InlineTimeDelta64{MINUTES}, "timedelta64[m]"),
+                (InlineTimeDelta64{(SECONDS, 5)}, "timedelta64[5s]"),
+                (InlineTimeDelta64{NumpyDates.UNBOUND_UNITS}, "timedelta64"),
             ]
                 @test pyeq(Bool, pygetattr(pyjl(t), "__numpy_dtype__"), np.dtype(d))
                 @test pyeq(Bool, np.dtype(pyjl(t)), np.dtype(d))
             end
 
             # unsupported cases
             @testset "$t -> AttributeError" for t in [
-                ComplexF16,
                 String,
-                Tuple{},
+                Vector{Int},
+                Ptr{Cvoid},
                 Ptr{Int},
                 Ptr{PythonCall.C.PyPtr},
             ]