removed import of Sage in test.py -KL