Skip to content

Commit

Permalink
Improve dcgmi test
Browse files Browse the repository at this point in the history
  • Loading branch information
Deezzir committed Sep 12, 2024
1 parent 36a913f commit 558007e
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
2 changes: 1 addition & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -47,4 +47,4 @@ functional: build
@TEST_SNAP=${SNAP_FILE} tox -e func -- ${FUNC_ARGS}

# The targets below don't depend on a file
.PHONY: help clean build lint reformat unittests functional test
.PHONY: help clean build lint reformat functional test
3 changes: 1 addition & 2 deletions tests/functional/test_snap_dcgm.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,7 @@ def test_dcgm_nv_hostengine():

def test_dcgmi():
"""Test of the dcgmi command."""
result = subprocess.run(["dcgm.dcgmi", "discovery", "-l"], capture_output=True, text=True)
print(result)
result = subprocess.run(["dcgm.dcgmi", "discovery", "-l"], check=True, capture_output=True, text=True)
assert "GPU ID" in result.stdout.strip(), "DCGMI is not working"


Expand Down

0 comments on commit 558007e

Please sign in to comment.