add assertions to predict
Merge branch 'work'