Update tests to reflect changes in the Python exposure