Fixes a bug.

aryann · Oct 6, 2024 · d94e8d8 · d94e8d8
1 parent 4be5d92
commit d94e8d8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/engine/src/moves.test.ts b/engine/src/moves.test.ts
@@ -70,6 +70,18 @@ describe("pawns", () => {
     assert.sameDeepMembers(generator.generateMoves("g6"), ["f7", "h7"]);
     assert.sameDeepMembers(generator.generateMoves("h6"), ["g7"]);
   });
+
+  it("all black pawns on rank 4", () => {
+    const generator = make("rnbqkbnr/8/8/8/pppppppp/8/PPPPPPPP/RNBQKBNR");
+    assert.sameDeepMembers(generator.generateMoves("a2"), ["a3"]);
+    assert.sameDeepMembers(generator.generateMoves("b2"), ["b3"]);
+    assert.sameDeepMembers(generator.generateMoves("c2"), ["c3"]);
+    assert.sameDeepMembers(generator.generateMoves("d2"), ["d3"]);
+    assert.sameDeepMembers(generator.generateMoves("e2"), ["e3"]);
+    assert.sameDeepMembers(generator.generateMoves("f2"), ["f3"]);
+    assert.sameDeepMembers(generator.generateMoves("g2"), ["g3"]);
+    assert.sameDeepMembers(generator.generateMoves("h2"), ["h3"]);
+  });
 });
 
 describe("queens", () => {

diff --git a/engine/src/moves.ts b/engine/src/moves.ts
@@ -128,7 +128,7 @@ export class MoveGenerator {
           const secondFile = frontFile + front.file;
           const secondRank = frontRank + front.rank;
           const secondSquare = SQUARES[this.toIndex(secondFile, secondRank)];
-          const secondDestination = this.board.get(frontSquare);
+          const secondDestination = this.board.get(secondSquare);
           if (!secondDestination) {
             moves.push(secondSquare);
           }