Added follow-up inference test cases.