Add Python test support