Add another unit test to ensure correctness of pixel attention mask placement

xenova · xenova · commit cadf68928550 · 2024-12-02T16:47:26.000Z
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);
         }
 
         {

Original file line number	Diff line number	Diff line change
`@@ -711,6 +711,9 @@ describe("Processors", () => {`
`711`	`711`	`);`
`712`	`712`	`compare(rows, [[0], [0], [0, 0]]);`
`713`	`713`	`compare(cols, [[0], [0], [0, 0]]);`
	`714`	`+`
	`715`	`+ // Test that the order of the pixel attention mask matches the python implementation`
	`716`	`+ compare(pixel_attention_mask.data.reduce((a, b, i) => a + i * b, 0), 228217205216);`
`714`	`717`	`}`
`715`	`718`
`716`	`719`	`{`