@@ -1432,29 +1432,48 @@ test_long_api(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(args))
1432
1432
PyLongWriter * writer ;
1433
1433
static PyLongExport long_export ;
1434
1434
1435
- writer = PyLongWriter_Create (1 , 1 , (void * * )& digits );
1435
+ writer = PyLongWriter_Create (1 , 1 , (void * * )& digits );
1436
+ if (writer == NULL ) {
1437
+ return NULL ;
1438
+ }
1436
1439
PyLongWriter_Discard (writer );
1437
1440
1438
- writer = PyLongWriter_Create (1 , 1 , (void * * )& digits );
1441
+ writer = PyLongWriter_Create (1 , 1 , (void * * )& digits );
1442
+ if (writer == NULL ) {
1443
+ return NULL ;
1444
+ }
1439
1445
digits [0 ] = 123 ;
1440
1446
obj = PyLongWriter_Finish (writer );
1447
+ if (obj == NULL ) {
1448
+ return NULL ;
1449
+ }
1441
1450
1442
1451
check_int (obj , -123 );
1443
- PyLong_Export (obj , & long_export );
1452
+ if (PyLong_Export (obj , & long_export ) < 0 ) {
1453
+ return NULL ;
1454
+ }
1444
1455
assert (long_export .value == -123 );
1445
1456
assert (long_export .digits == NULL );
1446
1457
PyLong_FreeExport (& long_export );
1447
1458
Py_DECREF (obj );
1448
1459
1449
- writer = PyLongWriter_Create (0 , 5 , (void * * )& digits );
1460
+ writer = PyLongWriter_Create (0 , 5 , (void * * )& digits );
1461
+ if (writer == NULL ) {
1462
+ return NULL ;
1463
+ }
1450
1464
digits [0 ] = 1 ;
1451
1465
digits [1 ] = 0 ;
1452
1466
digits [2 ] = 0 ;
1453
1467
digits [3 ] = 0 ;
1454
1468
digits [4 ] = 1 ;
1455
1469
obj = PyLongWriter_Finish (writer );
1470
+ if (obj == NULL ) {
1471
+ return NULL ;
1472
+ }
1456
1473
1457
- PyLong_Export (obj , & long_export );
1474
+ if (PyLong_Export (obj , & long_export ) < 0 ) {
1475
+ return NULL ;
1476
+ }
1458
1477
assert (long_export .value == 0 );
1459
1478
digits = (digit * )long_export .digits ;
1460
1479
assert (digits [0 ] == 1 );
0 commit comments