1 files changed, 15 insertions, 2 deletions
diff --git a/infra/libkookie/nixpkgs/pkgs/development/libraries/science/math/libtorch/test/test.cpp b/infra/libkookie/nixpkgs/pkgs/development/libraries/science/math/libtorch/test/test.cpp
index 5537724ce5f6..ca238fba521d 100644
--- a/infra/libkookie/nixpkgs/pkgs/development/libraries/science/math/libtorch/test/test.cpp
+++ b/infra/libkookie/nixpkgs/pkgs/development/libraries/science/math/libtorch/test/test.cpp
@@ -1,7 +1,20 @@
-#include <torch/torch.h>
+#undef NDEBUG
+#include <cassert>
+
 #include <iostream>
 
+#include <torch/torch.h>
+
 int main() {
   torch::Tensor tensor = torch::eye(3);
-  std::cout << tensor << std::endl;
+
+  float checkData[] = {
+    1, 0, 0,
+    0, 1, 0,
+    0, 0, 1
+  };
+
+  torch::Tensor check = torch::from_blob(checkData, {3, 3});
+
+  assert(tensor.allclose(check));
 }