How to Mock H2O in Py.test