refactored the test class to use unittest and moved the test script into the tests folder