Add test for python