[V2 Pipeline] SImple Asyncio pipeline test

setup.py

@@ -107,6 +107,7 @@ def _parse_requirements_file(file_path):
     "flask>=1.0.0",
     "flask-cors>=3.0.0",
     "Pillow>=8.3.2",
+    "pytest-asyncio==0.23.2",
 ]
 _docs_deps = [
     "m2r2~=0.2.7",

tests/deepsparse/pipelines/test_basic_pipeline.py

@@ -20,10 +20,12 @@
 
 from pydantic import BaseModel
 
+import pytest
 from deepsparse import Pipeline
 from deepsparse.operators import Operator
 from deepsparse.routers import LinearRouter
 from deepsparse.schedulers import OperatorScheduler
+from deepsparse.utils import InferenceState
 
 
 class IntSchema(BaseModel):
@@ -58,3 +60,16 @@ def test_run_simple_pipeline():
     pipeline_output = AddThreePipeline(pipeline_input)
 
     assert pipeline_output.value == 8
+
+
+@pytest.mark.asyncio
+async def test_run_async_simple_pipeline():
+    inference_state = InferenceState()
+    inference_state.create_state({})
+    pipeline_input = IntSchema(value=5)
+
+    pipeline_output = await AddThreePipeline.run_async(
+        pipeline_input, inference_state=inference_state
+    )
+
+    assert pipeline_output.value == 8