add linear regression unit test