add my own random test to demo.py