Fixed tests to actually work. Did not match unittest conventions.