test

usegalaxy-au · May 28, 2024 · d7e8773 · d7e8773
1 parent 7b8e358
commit d7e8773
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/tools/gpu_tools_test/nvidia-container-cli-info.xml b/tools/gpu_tools_test/nvidia-container-cli-info.xml
@@ -6,14 +6,41 @@
         <container type="docker">nvcr.io/nvidia/k8s/container-toolkit:v@[email protected]</container>
     </requirements>
     <command detect_errors="exit_code"><![CDATA[
-    nvidia-container-cli info
+    nvidia-container-cli info 2>&1 | tee out.txt
     ]]></command>
     <inputs>
     </inputs>
     <outputs>
         <data format="txt" name="out_txt" label="Output from nvidia-container-cli info" from_work_dir="out.txt" />
     </outputs>
+    <tests>
+    <test expect_num_outputs="1">
+        <output name="out_txt">
+            <assert_contents>
+                <has_text text="NVRM version"/>
+                <has_text text="CUDA version"/>
+            </assert_contents>
+        </output>
+    </test>
+
+    </tests>
     <help><![CDATA[
-    TODO: Fill in help.
+Just runs ``nvidia-container-cli info``. If you have planemo configured
+properly you will see output like this:
+
+.. code:: bash
+
+   NVRM version:   525.147.05
+   CUDA version:   12.0
+
+   Device Index:   0
+   Device Minor:   0
+   Model:          NVIDIA A100-PCIE-40GB
+   Brand:          Nvidia
+   GPU UUID:       GPU-3a29f0dc-490e-1e8d-abf3-a5cfa02adcde
+   Bus Location:   00000000:00:08.0
+   Architecture:   8.0
+
+
     ]]></help>
 </tool>