Update the linear regression tests to use the new data structure.