be robust against test being run over and over (such as -R)

This commit is contained in:
Benjamin Peterson 2010-01-30 23:28:38 +00:00
parent a04ae012ce
commit 46bff79d1f

View file

@ -1144,13 +1144,23 @@ raise_exception(PyObject *self, PyObject *args)
return NULL; return NULL;
} }
static int test_run_counter = 0;
static PyObject * static PyObject *
test_datetime_capi(PyObject *self, PyObject *args) { test_datetime_capi(PyObject *self, PyObject *args) {
if (PyDateTimeAPI) { if (PyDateTimeAPI) {
PyErr_SetString(PyExc_AssertionError, if (test_run_counter) {
"PyDateTime_CAPI somehow initialized"); /* Probably regrtest.py -R */
return NULL; Py_RETURN_NONE;
}
else {
PyErr_SetString(PyExc_AssertionError,
"PyDateTime_CAPI somehow initialized");
return NULL;
}
} }
test_run_counter++;
PyDateTime_IMPORT; PyDateTime_IMPORT;
if (PyDateTimeAPI) if (PyDateTimeAPI)
Py_RETURN_NONE; Py_RETURN_NONE;