From d7e8773e68c00ca870720d3068440a687e463f04 Mon Sep 17 00:00:00 2001
From: Tom Harrop <tharrop@unimelb.edu.au>
Date: Wed, 29 May 2024 09:05:30 +1000
Subject: [PATCH] test

---
 .../nvidia-container-cli-info.xml             | 31 +++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/tools/gpu_tools_test/nvidia-container-cli-info.xml b/tools/gpu_tools_test/nvidia-container-cli-info.xml
index a1d50a97..0fa14d4d 100644
--- 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@TOOL_VERSION@-ubuntu20.04</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>
\ No newline at end of file