Made test case for case in tests.py