This commit is contained in:
blueloveTH 2024-09-22 21:47:03 +08:00
parent c55dcd9ab8
commit 59ec34b2a1
3 changed files with 7 additions and 7 deletions

View File

@ -1,12 +1,10 @@
import sys
import math import math
from typing import TYPE_CHECKING
is_pkpy = not hasattr(sys, 'getrefcount') if TYPE_CHECKING:
if is_pkpy:
import numpy_bindings as np
else:
import numpy as np import numpy as np
else:
np = __import__('E:/pocketpy/3rd/numpy/build/Release/numpy.dll')
def assert_equal(a, b): def assert_equal(a, b):
assert (a == b).all() assert (a == b).all()
@ -888,3 +886,5 @@ assert_equal(arr1[-2:3:1], np.array([[[1, 2, 3, 4, 5],
[6, 7, 8, 9, 10]], [6, 7, 8, 9, 10]],
[[11, 12, 13, 14, 15], [[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20]]])) [16, 17, 18, 19, 20]]]))
print("ALL TESTS PASSED")

View File

@ -3,7 +3,6 @@
// clang-format off // clang-format off
#if defined(_WIN32) || defined(_WIN64) #if defined(_WIN32) || defined(_WIN64)
// define something for Windows (32-bit and 64-bit, this part is common)
#ifdef PY_DYNAMIC_MODULE #ifdef PY_DYNAMIC_MODULE
#define PK_API __declspec(dllimport) #define PK_API __declspec(dllimport)
#else #else

View File

@ -55,6 +55,7 @@ using module_ = module;
extern "C" PK_EXPORT bool py_module_initialize() { \ extern "C" PK_EXPORT bool py_module_initialize() { \
auto m = ::pkbind::module::create(#name); \ auto m = ::pkbind::module::create(#name); \
_pkbind_register_##name(m); \ _pkbind_register_##name(m); \
py_assign(py_retval(), m.ptr()); \
return true; \ return true; \
} \ } \
static void _pkbind_register_##name(::pkbind::module& variable) static void _pkbind_register_##name(::pkbind::module& variable)