May 5, 2025 · Mar 1, 2025 · Mar 1, 2025 · Mar 1, 2025 · Mar 1, 2025 · Mar 1, 2025
diff --git a/Lib/test/test_heapq.py b/Lib/test/test_heapq.py
 # _heapq.nlargest/nsmallest are saved in heapq._nlargest/_smallest when
 # _heapq is imported, so check them there
 func_names = ['heapify', 'heappop', 'heappush', 'heappushpop', 'heapreplace',
              'heappop_max', 'heapreplace_max', 'heapify_max', 'heappushpop_max']
              'heapify_max', 'heappop_max', 'heappush_max', 'heappushpop_max',
              'heapreplace_max']

 class TestModules(TestCase):
    def test_py_functions(self):
    @skipUnless(c_heapq, 'requires _heapq')
    def test_c_functions(self):
        for fname in func_names:
            self.assertEqual(getattr(c_heapq, fname).__module__, '_heapq')
            self.assertEqual(getattr(c_heapq, fname).__module__, '_heapq', fname)


 def load_tests(loader, tests, ignore):
diff --git a/Modules/_heapqmodule.c b/Modules/_heapqmodule.c
    return siftdown_max(heap, startpos, pos);
 }

 /*[clinic input]
 _heapq.heappush_max

    heap: object(subclass_of='&PyList_Type')
    item: object
    /

 Push item onto max heap, maintaining the heap invariant.
 [clinic start generated code]*/

 static PyObject *
 _heapq_heappush_max_impl(PyObject *module, PyObject *heap, PyObject *item)
 /*[clinic end generated code: output=c869d5f9deb08277 input=4743d7db137b6e2b]*/
 {
    if (PyList_Append(heap, item)) {
        return NULL;
    }

    if (siftdown_max((PyListObject *)heap, 0, PyList_GET_SIZE(heap)-1)) {
        return NULL;
    }

    Py_RETURN_NONE;
 }

 /*[clinic input]
 _heapq.heappop_max
    _HEAPQ_HEAPPOP_METHODDEF
    _HEAPQ_HEAPREPLACE_METHODDEF
    _HEAPQ_HEAPIFY_METHODDEF

    _HEAPQ_HEAPPUSH_MAX_METHODDEF
    _HEAPQ_HEAPPUSHPOP_MAX_METHODDEF
    _HEAPQ_HEAPPOP_MAX_METHODDEF
    _HEAPQ_HEAPIFY_MAX_METHODDEF
    _HEAPQ_HEAPREPLACE_MAX_METHODDEF
    _HEAPQ_HEAPPUSHPOP_MAX_METHODDEF
    _HEAPQ_HEAPIFY_MAX_METHODDEF

    {NULL, NULL}           /* sentinel */
 };

diff --git a/Modules/clinic/_heapqmodule.c.h b/Modules/clinic/_heapqmodule.c.h
Original file line number	Diff line number	Diff line change
Expand Up		@@ -14,7 +14,8 @@
		# _heapq.nlargest/nsmallest are saved in heapq._nlargest/_smallest when
		# _heapq is imported, so check them there
		func_names = ['heapify', 'heappop', 'heappush', 'heappushpop', 'heapreplace',
		'heappop_max', 'heapreplace_max', 'heapify_max', 'heappushpop_max']
		'heapify_max', 'heappop_max', 'heappush_max', 'heappushpop_max',
		'heapreplace_max']
StanFromIreland marked this conversation as resolved. OutdatedShow resolvedHide resolved

		class TestModules(TestCase):
		def test_py_functions(self):
Expand All		@@ -24,7 +25,7 @@ def test_py_functions(self):
		@skipUnless(c_heapq, 'requires _heapq')
		def test_c_functions(self):
		for fname in func_names:
		self.assertEqual(getattr(c_heapq, fname).__module__, '_heapq')
		self.assertEqual(getattr(c_heapq, fname).__module__, '_heapq', fname)


		def load_tests(loader, tests, ignore):
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -480,6 +480,30 @@ siftup_max(PyListObject *heap, Py_ssize_t pos)
		return siftdown_max(heap, startpos, pos);
		}

		/*[clinic input]
		_heapq.heappush_max

		heap: object(subclass_of='&PyList_Type')
		item: object
		/

		Push item onto max heap, maintaining the heap invariant.
		[clinic start generated code]*/

		static PyObject *
		_heapq_heappush_max_impl(PyObject module, PyObject heap, PyObject *item)
		/[clinic end generated code: output=c869d5f9deb08277 input=4743d7db137b6e2b]/
		{
		if (PyList_Append(heap, item)) {
		return NULL;
		}

		if (siftdown_max((PyListObject *)heap, 0, PyList_GET_SIZE(heap)-1)) {
		return NULL;
		}

		Py_RETURN_NONE;
		}

		/*[clinic input]
		_heapq.heappop_max
Expand DownExpand Up		@@ -585,10 +609,13 @@ static PyMethodDef heapq_methods[] = {
		_HEAPQ_HEAPPOP_METHODDEF
		_HEAPQ_HEAPREPLACE_METHODDEF
		_HEAPQ_HEAPIFY_METHODDEF

		_HEAPQ_HEAPPUSH_MAX_METHODDEF
		_HEAPQ_HEAPPUSHPOP_MAX_METHODDEF
		_HEAPQ_HEAPPOP_MAX_METHODDEF
		_HEAPQ_HEAPIFY_MAX_METHODDEF
		_HEAPQ_HEAPREPLACE_MAX_METHODDEF
		_HEAPQ_HEAPPUSHPOP_MAX_METHODDEF
		_HEAPQ_HEAPIFY_MAX_METHODDEF

		{NULL, NULL} /* sentinel */
		};

Expand Down