remove prints from test.py