Skip to content

Commit 7f96b08

Browse files
committed
feat: add unit test
Signed-off-by: Mehant Kammakomati <[email protected]>
1 parent 1e22a13 commit 7f96b08

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

plugins/mamba-cp/tests/test_mcp_plugin.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,8 @@
1616
import os
1717

1818
# Third Party
19-
from fms_acceleration.utils import instantiate_framework, read_configuration
2019
import pytest
2120

22-
# First Party
23-
from fms_acceleration_mcp import MCPAccelerationPlugin
24-
2521
# configuration
2622
DIRNAME = os.path.dirname(__file__)
2723
CONFIG_PATH = os.path.join(DIRNAME, "../configs/mcp.yaml")
@@ -32,6 +28,14 @@
3228
reason="mamba_ssm is not installed",
3329
)
3430
def test_framework_installs_mcp_plugin():
31+
# Third Party
32+
# pylint: disable=import-outside-toplevel
33+
from fms_acceleration.utils import instantiate_framework, read_configuration
34+
35+
# First Party
36+
# pylint: disable=import-outside-toplevel
37+
from fms_acceleration_mcp import MCPAccelerationPlugin
38+
3539
with instantiate_framework(
3640
read_configuration(CONFIG_PATH), require_packages_check=False
3741
) as framework:

0 commit comments

Comments
 (0)