Got test rewritten with unittest