Add unittest python script