Updating the test cases

Looks like one version of the tests are complete. For now this might be
good to go! :ship: :+1:

ps: emoji credited to @raghothams