Files
cpy3/subinterp_test.go
zeekayandGitHub aa54594e88 Hanzo fork v0.2.1: Python 3.12+ compat, sub-interp wrapper, free-threading detection, JSON hot-path + 3.13t build script (#1)
* feat: Python 3.12+/3.13 compat + sub-interpreter wrapper

Hanzo fork of go-python/cpy3. See HANZO_CHANGES.md for full diff.

- Polyfill APIs removed in 3.12 (per-type ClearFreeList helpers) and
  3.13 (PyEval_InitThreads, Py_SetPath/PythonHome/ProgramName,
  PySys_SetArgv/AddWarnOption/AddXOption/SetPath,
  Py_SetStandardStreamEncoding, PyEval_ThreadsInitialized) so existing
  call sites compile against modern CPython. Polyfills route to the
  PyConfig API where there's a meaningful replacement, return 0/no-op
  otherwise.

- New PEP 684 SubInterpreter wrapper (subinterp.go) — recommended
  primitive for tenant/per-worker isolation on 3.12+. Defaults follow
  PEP 684 (own GIL, threads on, no daemon threads, refuse non-isolated
  C extensions). Falls back to Py_NewInterpreter on 3.11 and older.

- Free-threading awareness (PEP 703) — IsGILDisabled() reports at
  runtime whether the linked CPython was built with --disable-gil.

- pkg-config switched from "python3" to "python3-embed" so libpython
  links cleanly on Linux distros that split the embed flags.

- Module path: github.com/go-python/cpy3 -> github.com/hanzoai/cpy3.
  Go 1.21+. Package name (python3) unchanged.

- CI matrix: Python {3.11, 3.12, 3.13} x Go {1.21, 1.22} on Ubuntu.

License remains MIT. Original go-python team credited in HANZO_CHANGES.md
and every modified file's header.

* feat: JSON hot-path helper + 3.13t build script

- LoadSource + CallJSONFunctionByName: single-cgo-crossing helpers
  for the common "JSON-in, JSON-out" extension embedding contract.
  Roughly 2x faster than the long form via individual wrappers and
  stack-safer under sustained load (macOS SIGBUS at ~400k iters of
  the 9-call long form is mitigated by collapsing to 1 cgo call).
- ErrUnknownFunction: sentinel error for missing/non-callable
  lookups so callers can errors.Is.
- scripts/build-python313t.sh: idempotent local build of CPython
  3.13.x with --disable-gil. Writes a python3-embed.pc so callers
  steer via PKG_CONFIG_PATH.
- HANZO_CHANGES.md + README updated.
- .gitignore: exclude python313t-build/ python313t-src/.

* ci: split test runs by Python global state

TestSubInterpreter_* and TestCallJSONFunctionByName_* / TestLoadSource_*
share global Py state with the data-type/error tests. Running them
in the same `go test` invocation can hang on darwin after the
data-type tests have already touched the main interpreter. Split
into two test steps so each starts from a fresh process.

* test: move json_hotpath tests in-package + share Py init

Move from package python3_test to package python3 so they share
Py_Initialize() state with subinterp_test.go. Each test pins its
own OS thread, calls Py_Initialize (idempotent), creates a fresh
sub-interp, and unpins on return. No PyEval_SaveThread in teardown
— the sub-interp's Close already released the calling thread's GIL.

This eliminates a SIGSEGV-on-second-test pattern caused by trying
to PyEval_SaveThread a state that's already been saved.

* docs: README example for the JSON hot path

Show LoadSource + CallJSONFunctionByName end-to-end with luxfi/log
for context. The helpers themselves take no logger dependency.
2026-05-18 23:38:11 -07:00

56 lines
1.5 KiB
Go

/*
SPDX-License-Identifier: MIT
Hanzo additions, 2026-05-19.
Sub-interpreter wrapper tests.
*/
package python3
import (
gort "runtime"
"testing"
)
func TestIsGILDisabled_RuntimeQuery(t *testing.T) {
// Just make sure it doesn't blow up. Value depends on whether
// the linked CPython was built with --disable-gil. We log it so
// CI logs show which build we're on.
v := IsGILDisabled()
t.Logf("IsGILDisabled = %v (Python version = %s)", v, Py_GetVersion())
}
// TestSubInterpreter_CreateAndClose verifies the smallest possible
// PEP 684 round trip — create a sub-interpreter with default config,
// enter it, run a trivial Python statement, leave, close.
//
// Requires Py_Initialize() to have run AND the calling thread to hold
// the main interpreter's GIL when we call NewSubInterpreter — the
// test harness handles that by pinning + initialising at the top.
func TestSubInterpreter_CreateAndClose(t *testing.T) {
gort.LockOSThread()
defer gort.UnlockOSThread()
Py_Initialize()
// Caller invariant: hold the main GIL before NewSubInterpreter.
// Py_Initialize leaves the calling thread holding the GIL.
si, err := NewSubInterpreter(DefaultSubInterpreterConfig())
if err != nil {
t.Fatalf("NewSubInterpreter: %v", err)
}
defer si.Close()
if si.ThreadState() == nil {
t.Fatal("ThreadState returned nil after successful create")
}
si.Enter()
// Trivial: run "1+1" as a Python expression via PyRun_SimpleString.
rc := PyRun_SimpleString("x = 1 + 1")
si.Leave()
if rc != 0 {
t.Fatalf("PyRun_SimpleString returned %d, want 0", rc)
}
}