added test.py
merged with testing