From 2245c9c5ea78f8579c72246b95f9125d90d16289 Mon Sep 17 00:00:00 2001
From: Artturin <Artturin@artturin.com>
Date: Wed, 29 Jan 2025 19:41:49 +0200
Subject: [PATCH] smi: Add test

---
 pkgs/smi/default.nix | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkgs/smi/default.nix b/pkgs/smi/default.nix
index b88382d..325e4b0 100644
--- a/pkgs/smi/default.nix
+++ b/pkgs/smi/default.nix
@@ -54,4 +54,10 @@ buildPythonPackage rec {
   ];
 
   dontUsePytestCheck = true; # no tests
+
+  installCheckPhase = ''
+    output=$($out/bin/tt-smi || true)
+    echo "tt-smi output: $output"
+    echo $output | grep -q "No Tenstorrent driver detected"
+  '';
 }