Unit tests for Python wrapper