Added tests for linear regression with grouping support