[test] update inference function to return also output mask, useful for tests (now on saturncloud test.ipynb notebook)

acbbf71