add test to learn python