I would put `[-1., -2.]` for the second example and check that we get a 0 prediction.