Create tests.py

test cases for ants game