Update state tests to use gocheck