Skip to content

Commit 1e22a13

Browse files
committed
feat: add unit test
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
1 parent 39162cb commit 1e22a13

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

plugins/mamba-cp/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ classifiers=[
2020
"Programming Language :: Python :: 3.11",
2121
]
2222

23-
dependencies = []
23+
dependencies = ["pytest"]
2424

2525
[tool.hatch.build.targets.wheel]
2626
only-include = ["src/fms_acceleration_mcp"]

plugins/mamba-cp/tests/test_mcp_plugin.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717

1818
# Third Party
1919
from fms_acceleration.utils import instantiate_framework, read_configuration
20+
import pytest
2021

2122
# First Party
2223
from fms_acceleration_mcp import MCPAccelerationPlugin
@@ -26,6 +27,10 @@
2627
CONFIG_PATH = os.path.join(DIRNAME, "../configs/mcp.yaml")
2728

2829

30+
@pytest.mark.skipif(
31+
not pytest.importorskip("mamba_ssm", reason="mamba_ssm is not installed"),
32+
reason="mamba_ssm is not installed",
33+
)
2934
def test_framework_installs_mcp_plugin():
3035
with instantiate_framework(
3136
read_configuration(CONFIG_PATH), require_packages_check=False

0 commit comments

Comments
 (0)