Added Python test