Added tests for python binding