validate instance type in test

Signed-off-by: Jeffrey Martin <[email protected]>
jmartin-tech · Apr 23, 2024 · baf17c0 · baf17c0
1 parent 56a2ce7
commit baf17c0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/generators/test_ggml.py b/tests/generators/test_ggml.py
@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-
 import os
 import pytest
 import tempfile
@@ -52,4 +50,4 @@ def test_init_good_model():
         file.write(garak.generators.ggml.GGUF_MAGIC)
         file.seek(0)
         g = garak.generators.ggml.GgmlGenerator(file.name)
-        assert g is not None
+        assert type(g) is garak.generators.ggml.GgmlGenerator