Add test case for python