Add more tests for the linear state updater.