Swtich from nose to unittest