updated test.pyx