Add tests for the LanguageModel Op

tests/text/test_model.py

@@ -0,0 +1,22 @@
+from outlines import string, compile
+from outlines.text.models.model import LanguageModel
+
+def test_initialize_model():
+
+    llm = LanguageModel(name="llm")
+
+    prompt = string()
+    out = llm(prompt)
+    assert isinstance(out.owner.op, LanguageModel)
+    assert out.owner.inputs[0] == prompt
+    assert out.name == "llm"
+
+
+class MockLM(LanguageModel):
+    def sample(self, _):
+        return "test"
+
+
+def test_sample():
+    llm = MockLM()
+    assert llm.perform("")[0] == "test"