Added import back into test_surrealism.py

Travis_CI build 121