On a similar note to the data generators, we can reuse this function from `LinearRegressionSuite` ?