More tests, implement infer() methods