Adding integration test that uses the sample preds