assert instead

iritkatriel · iritkatriel · commit eafd6b15f7a3 · 2023-10-19T16:44:14.000+01:00
diff --git a/Objects/genobject.c b/Objects/genobject.c
@@ -378,17 +378,19 @@ static PyObject *
 gen_close(PyGenObject *gen, PyObject *args)
 {
     PyObject *retval;
+    PyObject *yf = _PyGen_yf(gen);
     int err = 0;
 
     if (gen->gi_frame_state == FRAME_CREATED) {
+        assert(yf == NULL);
         gen->gi_frame_state = FRAME_COMPLETED;
         Py_RETURN_NONE;
     }
     if (gen->gi_frame_state >= FRAME_COMPLETED) {
+        assert(yf == NULL);
         Py_RETURN_NONE;
     }
 
-    PyObject *yf = _PyGen_yf(gen);
     if (yf) {
         PyFrameState state = gen->gi_frame_state;
         gen->gi_frame_state = FRAME_EXECUTING;

Original file line number	Diff line number	Diff line change
`@@ -378,17 +378,19 @@ static PyObject *`
`378`	`378`	`gen_close(PyGenObject gen, PyObject args)`
`379`	`379`	`{`
`380`	`380`	`PyObject *retval;`
	`381`	`+ PyObject *yf = _PyGen_yf(gen);`
`381`	`382`	`int err = 0;`
`382`	`383`
`383`	`384`	`if (gen->gi_frame_state == FRAME_CREATED) {`
	`385`	`+ assert(yf == NULL);`
`384`	`386`	`gen->gi_frame_state = FRAME_COMPLETED;`
`385`	`387`	`Py_RETURN_NONE;`
`386`	`388`	`}`
`387`	`389`	`if (gen->gi_frame_state >= FRAME_COMPLETED) {`
	`390`	`+ assert(yf == NULL);`
`388`	`391`	`Py_RETURN_NONE;`
`389`	`392`	`}`
`390`	`393`
`391`		`- PyObject *yf = _PyGen_yf(gen);`
`392`	`394`	`if (yf) {`
`393`	`395`	`PyFrameState state = gen->gi_frame_state;`
`394`	`396`	`gen->gi_frame_state = FRAME_EXECUTING;`