def test_score_dataset_with_ensemble_gives_same_predictions_as_score_dataset(
self):
# We're just going to test something simple here: that the methods don't crash, and that we
# get the same result with an ensemble of one model that we do with `score_dataset`.
run_model(self.param_path)
predictions, _ = score_dataset(self.param_path, [self.TEST_FILE])
ensembled_predictions, _ = score_dataset_with_ensemble(
[self.param_path], [self.TEST_FILE])
assert_almost_equal(predictions, ensembled_predictions)
def test_score_dataset(self):
run_model(self.param_path)
score_dataset(self.param_path)
def test_score_dataset_does_not_crash(self):
run_model(self.param_path)
score_dataset(self.param_path, [self.TEST_FILE])