added more tests to (python)