Create shapes test.py