Merge branch 'main' into wma/heuristic

finch-tensor · May 19, 2024 · f15b948 · f15b948
2 parents d4f691f + 8c107d1
commit f15b948
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/interface/fileio/binsparse.jl b/src/interface/fileio/binsparse.jl
@@ -313,7 +313,7 @@ function bspread_header end
 function bspread(f)
     desc = bspread_header(f)["binsparse"]
     @assert desc["version"] == "$BINSPARSE_VERSION"
-    fmt = OrderedDict{Any, Any}(get(() -> desc["tensor"], bspread_tensor_lookup, desc["format"]))
+    fmt = OrderedDict{Any, Any}(get(() -> bspread_tensor_lookup[desc["format"]], desc, "tensor"))
     if !haskey(fmt, "transpose")
         fmt["transpose"] = collect(0:length(desc["shape"]) - 1)
     end

diff --git a/test/test_fileio.jl b/test/test_fileio.jl
@@ -53,6 +53,13 @@ using Pkg
                     end
                 end
             end
+
+            B = fsprand(100, 100, 100, 0.1)
+            @testset "binsparse COO3" begin
+                fname = joinpath(f, "foo.bsp.h5")
+                bspwrite(fname, B)
+                @test Structure(B) == Structure(bspread(fname))
+            end
         end
     end