Create ex_test.py

This is the unit test for our project