iterators and fill ops

1 files changed, 13 insertions, 0 deletions

diff --git a/modules/codec/tests/math.cpp b/modules/codec/tests/math.cpp
index 63b6d2b..5f4de92 100644
--- a/modules/codec/tests/math.cpp
+++ b/modules/codec/tests/math.cpp
@@ -28,6 +28,19 @@ SAW_TEST("Math/Tensor Construction"){
 	// data<sch::Vector<sch::Int32,3u>> b{{{1,3,5}}};
 }
 
+SAW_TEST("Math/Tensor Fill"){
+	using namespace saw;
+
+	auto tensor = math::fill<sch::Float64,2u,2u>({2.5});
+
+	bool cf{true};
+	for(uint64_t i = 0u; i < 2u; ++i){
+		for(uint64_t j = 0u; j < 2u; ++j){
+			SAW_EXPECT((tensor.at({{i,j}}).get() == 2.5), "Wasn't filled with value 2.5");
+		}
+	}
+}
+
 SAW_TEST("Math/Tensor"){
 	using namespace saw;