Trying unittest