Updated unit testing with input tests