reduce to a simple test (failing)

jafioti · Jun 20, 2024 · dddd366 · dddd366
1 parent adfc4ac
commit dddd366
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 72 deletions.
diff --git a/examples/bias_test/Cargo.toml b/examples/bias_test/Cargo.toml
diff --git a/examples/bias_test/src/main.rs b/examples/bias_test/src/main.rs
diff --git a/src/hl_ops/movement.rs b/src/hl_ops/movement.rs
@@ -514,4 +514,14 @@ mod tests {
 
         assert_close(&c.data(), &d_c.as_vec());
     }
+
+    #[test]
+    fn test_noop_expand() {
+        type S = R1<1>;
+        type Tensor = GraphTensor<S>;
+        let mut cx = Graph::new();
+        let a: Tensor = cx.tensor();
+        let noop_expanded: Tensor = a.expand::<S, LAxis<0>>();
+        assert_eq!(a.shape, noop_expanded.shape);
+    }
 }