Adding test file for python wrapper and struggling against type errors

Change-Id: If193c341cc60b5769db64c4b244d184239727d8d