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
1 changed files with 13 additions and 3 deletions

View File

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