Added a test for inference.