init

transformers/tests/models/owlv2/test_processing_owlv2.py

@@ -0,0 +1,23 @@
+import shutil
+import tempfile
+import unittest
+
+from transformers import Owlv2Processor
+from transformers.testing_utils import require_scipy
+
+from ...test_processing_common import ProcessorTesterMixin
+
+
+@require_scipy
+class Owlv2ProcessorTest(ProcessorTesterMixin, unittest.TestCase):
+    processor_class = Owlv2Processor
+
+    @classmethod
+    def setUpClass(cls):
+        cls.tmpdirname = tempfile.mkdtemp()
+        processor = cls.processor_class.from_pretrained("google/owlv2-base-patch16-ensemble")
+        processor.save_pretrained(cls.tmpdirname)
+
+    @classmethod
+    def tearDownClass(cls):
+        shutil.rmtree(cls.tmpdirname, ignore_errors=True)