检查np.array是否与datetime类似的一致方法

Question

我要进行一些单元测试，我需要确保一个函数总是返回一个类似于nd.datetime64的对象。但是，它们可以是任何单位（年，日，纳秒等）。

我试过了：

comp = function_returns_datetime_array(inp)

assert isinstance(comp.dtype, np.datetime64)
assert issubclass(comp.dtype, np.datetime64)
assert issubclass(type(comp.dtype), np.datetime64)

有什么建议吗？

Answer 1

您可以使用issubdtype：

np.issubdtype(comp.dtype, np.datetime64)

Answer 2

目前我正在使用：

assert 'datetime64' in str(comp.dtype)

它在我的口中留下了不好的味道（因为它依赖于其他行为），但它是我所能满足的。