Unit testing with assert