Add another unit test to ensure correctness of pixel attention mask p…

…lacement
huggingface · Dec 2, 2024 · cadf689 · cadf689
1 parent 2e0c5bd
commit cadf689
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/processors.test.js b/tests/processors.test.js
@@ -711,6 +711,9 @@ describe("Processors", () => {
           );
           compare(rows, [[0], [0], [0, 0]]);
           compare(cols, [[0], [0], [0, 0]]);
+
+          // Test that the order of the pixel attention mask matches the python implementation
+          compare(pixel_attention_mask.data.reduce((a, b, i) => a + i * b, 0), 228217205216);
         }
 
         {