Create test.py

simple assert