test_subinterpreters.py

"""Tests that verify the C extension works in subinterpreters.

Subinterpreters became usable for third-party C extensions in
Python 3.12 (PEP 684). On 3.13+ the extension uses per-module state
and heap types so that each interpreter gets its own copy.
"""
import sys
import unittest
from unittest import TestCase

from simplejson.tests._helpers import skip_if_speedups_missing

@unittest.skipIf(sys.version_info < (3, 12),
                 "subinterpreters require Python 3.12+")
class TestSubinterpreters(TestCase):
    """Test that the C extension can be loaded in subinterpreters."""

def _run_in_subinterp(self, code):
        """Helper to run code in a fresh subinterpreter."""
        try:
            import _interpreters
        except ImportError:
            raise unittest.SkipTest("_interpreters not available")
        interp = _interpreters.create()
        try:
            _interpreters.run_string(interp, code)
        finally:
            _interpreters.destroy(interp)

@skip_if_speedups_missing
    def test_import_in_subinterpreter(self):
        """Verify _speedups can be imported in a subinterpreter."""
        self._run_in_subinterp(
            "import simplejson; simplejson.dumps({'a': 1})")

@skip_if_speedups_missing
    def test_encode_in_subinterpreter(self):
        """Verify encoding works correctly in a subinterpreter."""
        self._run_in_subinterp("""
import simplejson
assert simplejson.dumps(None) == 'null'
assert simplejson.dumps(True) == 'true'
assert simplejson.dumps(False) == 'false'
assert simplejson.dumps(42) == '42'
assert simplejson.dumps(3.14) == '3.14'
assert simplejson.dumps("hello") == '"hello"'
assert simplejson.dumps([1, 2, 3]) == '[1, 2, 3]'
assert simplejson.dumps({"a": 1}, sort_keys=True) == '{"a": 1}'
""")

@skip_if_speedups_missing
    def test_decode_in_subinterpreter(self):
        """Verify decoding works correctly in a subinterpreter."""
        self._run_in_subinterp("""
import simplejson
assert simplejson.loads('null') is None
assert simplejson.loads('true') is True
assert simplejson.loads('42') == 42
assert simplejson.loads('"hello"') == 'hello'
assert simplejson.loads('[1, 2, 3]') == [1, 2, 3]
assert simplejson.loads('{"a": 1}') == {"a": 1}
""")

@skip_if_speedups_missing
    def test_multiple_subinterpreters(self):
        """Verify multiple subinterpreters can use simplejson concurrently."""
        try:
            import _interpreters
        except ImportError:
            raise unittest.SkipTest("_interpreters not available")
        interps = [_interpreters.create() for _ in range(3)]
        try:
            for i, interp in enumerate(interps):
                _interpreters.run_string(interp, """
import simplejson
result = simplejson.dumps({"interp": %d})
assert '"interp": %d' in result
""" % (i, i))
        finally:
            for interp in interps:
                _interpreters.destroy(interp)

@skip_if_speedups_missing
    def test_subinterpreter_state_independent(self):
        """Verify destroying one subinterpreter doesn't affect another."""
        try:
            import _interpreters
        except ImportError:
            raise unittest.SkipTest("_interpreters not available")
        interp1 = _interpreters.create()
        interp2 = _interpreters.create()
        try:
            # Both interpreters load and use simplejson
            _interpreters.run_string(interp1,
                "import simplejson; simplejson.dumps([1])")
            _interpreters.run_string(interp2,
                "import simplejson; simplejson.dumps([2])")

# Destroy the first interpreter
            _interpreters.destroy(interp1)
            interp1 = None

# Second interpreter must still work correctly
            _interpreters.run_string(interp2, """
import simplejson
assert simplejson.dumps({"still": "works"}) == '{"still": "works"}'
assert simplejson.loads('{"still": "works"}') == {"still": "works"}
""")
        finally:
            if interp1 is not None:
                _interpreters.destroy(interp1)
            _interpreters.destroy(interp2)

@skip_if_speedups_missing
    @unittest.skipIf(sys.version_info < (3, 13),
                     "heap types require Python 3.13+")
    def test_subinterpreter_heap_types(self):
        """Verify types are heap types inside subinterpreters."""
        self._run_in_subinterp("""
from simplejson._speedups import make_scanner, make_encoder
# Py_TPFLAGS_HEAPTYPE = 1 << 9
assert make_scanner.__flags__ & (1 << 9), "Scanner should be heap type"
assert make_encoder.__flags__ & (1 << 9), "Encoder should be heap type"
""")

docroot: /home/induvedam/public_html

curdir: /opt/imunify360/venv/lib/python3.11/site-packages/simplejson/tests

	.	4096
	..	4096
X	__init__.py	2917
	__pycache__	4096
X	_cibw_runner.py	173
X	_helpers.py	405
X	test_bigint_as_string.py	2238
X	test_bitsize_int_as_string.py	4535
X	test_check_circular.py	1579
X	test_decimal.py	2544
X	test_decode.py	7470
X	test_default.py	221
X	test_dump.py	17600
X	test_encode_basestring_ascii.py	2337
X	test_encode_for_html.py	1515
X	test_errors.py	3503
X	test_fail.py	6454
X	test_float.py	1983
X	test_for_json.py	4126
X	test_free_threading.py	3169
X	test_indent.py	2568
X	test_item_sort_key.py	1376
X	test_iterable.py	2580
X	test_namedtuple.py	7391
X	test_pass1.py	1746
X	test_pass2.py	386
X	test_pass3.py	482
X	test_raw_json.py	1062
X	test_recursion.py	1679
X	test_scanstring.py	12098
X	test_separators.py	942
X	test_speedups.py	12590
X	test_str_subclass.py	740
X	test_subclass.py	1124
X	test_subinterpreters.py	4639
X	test_tool.py	2996
X	test_tuple.py	1831
X	test_unicode.py	7056