mirror of
https://github.com/lichess-org/mobile.git
synced 2026-05-26 13:50:52 +00:00
586 lines
30 KiB
Dart
586 lines
30 KiB
Dart
import 'package:collection/collection.dart';
|
|
import 'package:dartchess/dartchess.dart';
|
|
import 'package:fast_immutable_collections/fast_immutable_collections.dart';
|
|
import 'package:flutter_test/flutter_test.dart';
|
|
import 'package:lichess_mobile/src/model/common/chess.dart';
|
|
import 'package:lichess_mobile/src/model/common/eval.dart';
|
|
import 'package:lichess_mobile/src/model/common/node.dart';
|
|
import 'package:lichess_mobile/src/model/common/uci.dart';
|
|
|
|
void main() {
|
|
group('Node', () {
|
|
test('Root.fromPgnMoves', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
expect(root.position, equals(Chess.initial));
|
|
expect(root.children.length, equals(1));
|
|
final child = root.children.first;
|
|
expect(child.id, equals(UciCharPair.fromMove(Move.parse('e2e4')!)));
|
|
expect(child.sanMove, equals(SanMove('e4', Move.parse('e2e4')!)));
|
|
expect(
|
|
child.position.fen,
|
|
equals('rnbqkbnr/pppppppp/8/8/4P3/8/PPPP1PPP/RNBQKBNR b KQkq - 0 1'),
|
|
);
|
|
expect(child.position, equals(Chess.initial.playUnchecked(Move.parse('e2e4')!)));
|
|
});
|
|
|
|
test('Root.fromPgnGame, flat', () {
|
|
final pgnGame = PgnGame.parsePgn(fisherSpasskyPgn);
|
|
final root = Root.fromPgnGame(pgnGame);
|
|
expect(root.position, equals(Chess.initial));
|
|
expect(root.children.length, equals(1));
|
|
expect(root.mainline.length, equals(85));
|
|
});
|
|
|
|
test('Root.fromPgnGame, with variations', () {
|
|
const pgn = '1. e4 d5 2. exd5 Qxd5 (2... Nf6 3. c4 (3. Nc3)) 3. Nc3';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
expect(root.position, equals(Chess.initial));
|
|
expect(root.children.length, equals(1));
|
|
expect(root.mainline.length, equals(5));
|
|
final nodeWithVariation = root.nodeAt(UciPath.fromUciMoves(['e2e4', 'd7d5', 'e4d5']));
|
|
expect(nodeWithVariation.children.length, 2);
|
|
expect(nodeWithVariation.children[1].sanMove.san, equals('Nf6'));
|
|
expect(nodeWithVariation.children[1].children.length, 2);
|
|
});
|
|
|
|
test('nodesOn', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final path = UciPath.fromId(UciCharPair.fromUci('e2e4'));
|
|
final nodeList = root.nodesOn(path).toList();
|
|
expect(nodeList.length, equals(2));
|
|
expect(nodeList[0], equals(root));
|
|
expect(nodeList[1], equals(root.nodeAt(path) as Branch));
|
|
});
|
|
|
|
test('branchesOn, simple', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final path = UciPath.fromId(UciCharPair.fromUci('e2e4'));
|
|
final nodeList = root.branchesOn(path);
|
|
expect(nodeList.length, equals(1));
|
|
expect(nodeList.first, equals(root.nodeAt(path) as Branch));
|
|
|
|
expect(nodeList.map((n) => n.view), root.view.branchesOn(path));
|
|
});
|
|
|
|
test('branchesOn, with variation', () {
|
|
final root = Root.fromPgnMoves('e4 e5 Nf3');
|
|
final move = Move.parse('b1c3')!;
|
|
final (newPath, _) = root.addMoveAt(
|
|
UciPath.fromIds([UciCharPair.fromUci('e2e4'), UciCharPair.fromUci('e7e5')].lock),
|
|
move,
|
|
);
|
|
final newNode = root.nodeAt(newPath!);
|
|
|
|
// mainline has not changed
|
|
expect(root.mainline.length, equals(3));
|
|
expect(root.mainline.last, equals(root.nodeAt(root.mainlinePath) as Branch));
|
|
|
|
final nodeList = root.branchesOn(newPath);
|
|
expect(nodeList.length, equals(3));
|
|
expect(nodeList.last, equals(newNode));
|
|
|
|
expect(nodeList.map((n) => n.view), root.view.branchesOn(newPath));
|
|
});
|
|
|
|
test('mainline', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final mainline = root.mainline;
|
|
|
|
expect(mainline.length, equals(2));
|
|
final list = mainline.toList();
|
|
expect(list[0].sanMove, equals(SanMove('e4', Move.parse('e2e4')!)));
|
|
expect(list[1].sanMove, equals(SanMove('e5', Move.parse('e7e5')!)));
|
|
});
|
|
|
|
test('isOnMainline', () {
|
|
final root = Root.fromPgnMoves('e4 e5 Nf3');
|
|
final path = UciPath.fromId(UciCharPair.fromUci('e2e4'));
|
|
expect(root.isOnMainline(path), isTrue);
|
|
|
|
final move = Move.parse('b1c3')!;
|
|
final (newPath, _) = root.addMoveAt(
|
|
UciPath.fromIds([UciCharPair.fromUci('e2e4'), UciCharPair.fromUci('e7e5')].lock),
|
|
move,
|
|
);
|
|
|
|
expect(root.isOnMainline(newPath!), isFalse);
|
|
});
|
|
|
|
test('add child', () {
|
|
final root = Root(position: Chess.initial);
|
|
final child = Branch(sanMove: SanMove('e4', Move.parse('e2e4')!), position: Chess.initial);
|
|
root.addChild(child);
|
|
expect(root.children.length, equals(1));
|
|
expect(root.children.first, equals(child));
|
|
});
|
|
|
|
test('prepend child', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final child = Branch(sanMove: SanMove('d4', Move.parse('d2d4')!), position: Chess.initial);
|
|
root.prependChild(child);
|
|
expect(root.children.length, equals(2));
|
|
expect(root.children.first, equals(child));
|
|
expect(root.children.last.id, equals(UciCharPair.fromUci('e2e4')));
|
|
});
|
|
|
|
test('nodeAt', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = root.nodeAt(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
expect(branch, equals(root.children.first));
|
|
});
|
|
|
|
test('nodeAtOrNull', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = root.nodeAtOrNull(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
expect(branch, equals(root.children.first));
|
|
|
|
final branch2 = root.nodeAtOrNull(UciPath.fromId(UciCharPair.fromUci('b1c3')));
|
|
expect(branch2, isNull);
|
|
});
|
|
|
|
test('branchAt from root', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = root.branchAt(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
expect(branch, equals(root.children.first));
|
|
});
|
|
|
|
test('branchAt from branch', () {
|
|
final root = Root.fromPgnMoves('e4 e5 Nf3');
|
|
final branch = root.branchAt(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
final branch2 = branch!.branchAt(UciPath.fromId(UciCharPair.fromUci('e7e5')));
|
|
expect(branch2, equals(branch.children.first));
|
|
});
|
|
|
|
test('updateAt', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = Branch(sanMove: SanMove('Nc6', Move.parse('b8c6')!), position: Chess.initial);
|
|
|
|
final fromPath = UciPath.fromId(UciCharPair.fromUci('e2e4'));
|
|
final (nodePath, _) = root.addNodeAt(fromPath, branch);
|
|
|
|
expect(root.branchesOn(nodePath!), equals([root.children.first, branch]));
|
|
|
|
final eval = LocalEval(
|
|
position: branch.position,
|
|
searchTime: const Duration(seconds: 10),
|
|
cp: 100,
|
|
depth: 10,
|
|
nodes: 1000,
|
|
millis: 1000,
|
|
pvs: IList([
|
|
PvData(moves: IList(const ['e2e4'])),
|
|
]),
|
|
threatMode: false,
|
|
);
|
|
|
|
final newNode = root.updateAt(nodePath, (node) {
|
|
node.eval = eval;
|
|
});
|
|
|
|
expect(root.branchesOn(nodePath), equals([root.children.first, newNode!]));
|
|
});
|
|
|
|
test('updateAll', () {
|
|
final root = Root.fromPgnMoves('e4 e5 Nf3');
|
|
|
|
expect(root.mainline.map((n) => n.eval), equals([null, null, null]));
|
|
|
|
final eval = LocalEval(
|
|
position: root.position,
|
|
searchTime: const Duration(seconds: 10),
|
|
cp: 100,
|
|
depth: 10,
|
|
nodes: 1000,
|
|
millis: 1000,
|
|
pvs: IList([
|
|
PvData(moves: IList(const ['e2e4'])),
|
|
]),
|
|
threatMode: false,
|
|
);
|
|
|
|
root.updateAll((node) {
|
|
node.eval = eval;
|
|
});
|
|
|
|
expect(root.mainline.map((n) => n.eval), equals([eval, eval, eval]));
|
|
});
|
|
|
|
test('addNodeAt', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = Branch(sanMove: SanMove('Nc6', Move.parse('b8c6')!), position: Chess.initial);
|
|
final (newPath, isNewNode) = root.addNodeAt(
|
|
UciPath.fromId(UciCharPair.fromUci('e2e4')),
|
|
branch,
|
|
);
|
|
|
|
expect(
|
|
newPath,
|
|
equals(UciPath.fromIds(IList([UciCharPair.fromUci('e2e4'), UciCharPair.fromUci('b8c6')]))),
|
|
);
|
|
|
|
expect(isNewNode, isTrue);
|
|
|
|
final testNode = root.nodeAt(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
expect(testNode.children.length, equals(2));
|
|
expect(testNode.children[1], equals(branch));
|
|
});
|
|
|
|
test('addNodeAt, prepend', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = Branch(sanMove: SanMove('Nc6', Move.parse('b8c6')!), position: Chess.initial);
|
|
root.addNodeAt(UciPath.fromId(UciCharPair.fromUci('e2e4')), branch, prepend: true);
|
|
|
|
final testNode = root.nodeAt(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
expect(testNode.children.length, equals(2));
|
|
expect(testNode.children[0], equals(branch));
|
|
});
|
|
|
|
test('addNodeAt, with an existing node at path', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = Branch(sanMove: SanMove('e5', Move.parse('e7e5')!), position: Chess.initial);
|
|
final (newPath, isNewNode) = root.addNodeAt(
|
|
UciPath.fromId(UciCharPair.fromUci('e2e4')),
|
|
branch,
|
|
);
|
|
|
|
expect(
|
|
newPath,
|
|
equals(UciPath.fromIds(IList([UciCharPair.fromUci('e2e4'), UciCharPair.fromUci('e7e5')]))),
|
|
);
|
|
|
|
expect(isNewNode, isFalse);
|
|
|
|
final testNode = root.nodeAt(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
|
|
// node is not replaced
|
|
expect(testNode.children.length, equals(1));
|
|
expect(testNode.children[0], isNot(branch));
|
|
});
|
|
|
|
test('addNodesAt', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final branch = Branch(sanMove: SanMove('Nc6', Move.parse('b8c6')!), position: Chess.initial);
|
|
final branch2 = Branch(sanMove: SanMove('Na6', Move.parse('b8a6')!), position: Chess.initial);
|
|
root.addNodesAt(UciPath.fromId(UciCharPair.fromUci('e2e4')), [branch, branch2]);
|
|
|
|
final testNode = root.nodeAt(UciPath.fromId(UciCharPair.fromUci('e2e4')));
|
|
expect(testNode.children.length, equals(2));
|
|
expect(testNode.children[1], equals(branch));
|
|
expect(testNode.children[1].children[0], equals(branch2));
|
|
});
|
|
|
|
test('addMoveAt', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final move = Move.parse('b1c3')!;
|
|
final path = UciPath.fromIds([UciCharPair.fromUci('e2e4'), UciCharPair.fromUci('e7e5')].lock);
|
|
final currentPath = root.mainlinePath;
|
|
final (newPath, _) = root.addMoveAt(path, move);
|
|
expect(newPath, equals(currentPath + UciCharPair.fromMove(move)));
|
|
final newNode = root.branchAt(newPath!);
|
|
expect(newNode?.position.ply, equals(3));
|
|
expect(newNode?.sanMove, equals(SanMove('Nc3', move)));
|
|
expect(
|
|
newNode?.position.fen,
|
|
equals('rnbqkbnr/pppp1ppp/8/4p3/4P3/2N5/PPPP1PPP/R1BQKBNR b KQkq - 1 2'),
|
|
);
|
|
|
|
final testNode = root.nodeAt(path);
|
|
expect(testNode.children.length, equals(1));
|
|
expect(testNode.children.first.sanMove, equals(SanMove('Nc3', move)));
|
|
expect(
|
|
testNode.children.first.position.fen,
|
|
equals('rnbqkbnr/pppp1ppp/8/4p3/4P3/2N5/PPPP1PPP/R1BQKBNR b KQkq - 1 2'),
|
|
);
|
|
});
|
|
|
|
test('addMovesAt', () {
|
|
final root = Root.fromPgnMoves('e4 e5');
|
|
final moves = [Move.parse('b8c6')!, Move.parse('d2d4')!];
|
|
final newPath = root.addMovesAt(UciPath.fromId(UciCharPair.fromUci('e2e4')), moves);
|
|
|
|
expect(
|
|
newPath,
|
|
equals(
|
|
UciPath.fromIds([
|
|
UciCharPair.fromUci('e2e4'),
|
|
UciCharPair.fromUci('b8c6'),
|
|
UciCharPair.fromUci('d2d4'),
|
|
]),
|
|
),
|
|
);
|
|
|
|
final lastNewNode = root.branchAt(newPath!)!;
|
|
expect(lastNewNode.position.ply, 3);
|
|
expect(lastNewNode.sanMove, SanMove('d4', Move.parse('d2d4')!));
|
|
expect(
|
|
lastNewNode.position.fen,
|
|
'r1bqkbnr/pppppppp/2n5/8/3PP3/8/PPP2PPP/RNBQKBNR b KQkq - 0 2',
|
|
);
|
|
|
|
expect(root.children.first.children.length, 2);
|
|
});
|
|
|
|
test('deleteAt', () {
|
|
final root = Root.fromPgnMoves('e4 e5 Nf3');
|
|
final path = UciPath.fromIds([UciCharPair.fromUci('e2e4'), UciCharPair.fromUci('e7e5')]);
|
|
root.deleteAt(path);
|
|
expect(root.mainline.length, equals(1));
|
|
expect(root.mainline.last, equals(root.children.first));
|
|
});
|
|
|
|
test('deleteAt, root element', () {
|
|
final root = Root.fromPgnMoves('e4');
|
|
final path = UciPath.fromIds([UciCharPair.fromUci('e2e4')]);
|
|
root.deleteAt(path);
|
|
expect(root.mainline.length, equals(0));
|
|
});
|
|
|
|
test('promoteAt', () {
|
|
const pgn = '1. e4 d5 2. exd5 Qxd5 (2... Nf6 3. c4 (3. Nc3)) 3. Nc3';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
final path = UciPath.fromUciMoves(['e2e4', 'd7d5', 'e4d5', 'g8f6']);
|
|
expect(root.nodeAt(path), isNotNull);
|
|
root.promoteAt(path, toMainline: false);
|
|
expect(
|
|
root.mainline.map((n) => n.sanMove.san).toList(),
|
|
equals(['e4', 'd5', 'exd5', 'Nf6', 'c4']),
|
|
);
|
|
expect(
|
|
root.makePgn(),
|
|
equals('1. e4 d5 2. exd5 Nf6 ( 2... Qxd5 3. Nc3 ) 3. c4 ( 3. Nc3 ) *\n'),
|
|
);
|
|
});
|
|
|
|
test('promoteAt, to mainline', () {
|
|
const pgn = '1. e4 d5 2. exd5 Qxd5 (2... Nf6 3. c4 (3. Nc3)) 3. Nc3';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
final path = UciPath.fromUciMoves(['e2e4', 'd7d5', 'e4d5', 'g8f6', 'b1c3']);
|
|
expect(root.nodeAt(path), isNotNull);
|
|
root.promoteAt(path, toMainline: true);
|
|
expect(
|
|
root.mainline.map((n) => n.sanMove.san).toList(),
|
|
equals(['e4', 'd5', 'exd5', 'Nf6', 'Nc3']),
|
|
);
|
|
expect(
|
|
root.makePgn(),
|
|
equals('1. e4 d5 2. exd5 Nf6 ( 2... Qxd5 3. Nc3 ) 3. Nc3 ( 3. c4 ) *\n'),
|
|
);
|
|
});
|
|
|
|
test('promoteAt, root node', () {
|
|
const pgn = '1. e4 (1. d4)';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
final path = UciPath.fromUciMoves(['d2d4']);
|
|
expect(root.nodeAt(path), isNotNull);
|
|
root.promoteAt(path, toMainline: false);
|
|
expect(root.mainline.map((n) => n.sanMove.san).toList(), equals(['d4']));
|
|
expect(root.makePgn(), equals('1. d4 ( 1. e4 ) *\n'));
|
|
});
|
|
|
|
group('merge', () {
|
|
test('add moves', () {
|
|
const pgn = '''
|
|
1. d4 { [%clk 1:00:00] } Nf6 { [%clk 1:00:00] } 2. c4 { [%clk 1:00:00] } g6 { [%clk 1:00:00] } 3. Nc3 { [%clk 1:00:00] } Bg7 { [%clk 1:00:00] } 4. e4 { [%clk 1:00:00] } d6 { [%clk 1:00:00] } 5. f3 { [%clk 1:00:00] } O-O { [%clk 1:00:00] } 6. Be3 { [%clk 1:00:00] } e5 { [%clk 1:00:00] } 7. d5 { [%clk 1:00:00] } Nh5 { [%clk 1:00:00] } 8. Qd2 { [%clk 1:00:00] } Qh4+ { [%clk 1:00:00] } 9. g3 { [%clk 1:00:00] } Qe7 { [%clk 1:00:00] } 10. Nh3 { [%clk 1:00:00] } f5 { [%clk 0:56:44] } 11. exf5 { [%clk 0:58:18] } gxf5 { [%clk 0:55:20] } 12. O-O-O { [%clk 0:57:22] } Na6 { [%clk 0:52:30] } 13. Re1 { [%clk 0:52:22] } Nf6 { [%clk 0:48:20] } 14. Ng5 { [%clk 0:50:43] } c6 { [%clk 0:47:38] } 15. h4 { [%clk 0:50:01] } h6 { [%clk 0:46:10] } 16. Nh3 { [%clk 0:49:18] } cxd5 { [%clk 0:45:06] } 17. Bxh6 { [%clk 0:47:13] } Bxh6 { [%clk 0:44:17] } 18. Qxh6 { [%clk 0:45:59] } Bd7 { [%clk 0:43:34] } 19. cxd5 { [%clk 0:45:15] } Nc5 { [%clk 0:42:50] } 20. Kb1 { [%clk 0:44:14] } Qg7 { [%clk 0:41:29] } 21. Qd2 { [%clk 0:42:39] } e4 { [%clk 0:40:55] } 22. b4 { [%clk 0:40:31] } Na4 { [%clk 0:39:58] } 23. Nxa4 { [%clk 0:39:13] } Bxa4 { [%clk 0:38:39] } 24. Ng5 { [%clk 0:37:47] } Rfc8 { [%clk 0:37:14] } *
|
|
''';
|
|
|
|
const pgn2 = '''
|
|
1. d4 { [%clk 1:00:00] } Nf6 { [%clk 1:00:00] } 2. c4 { [%clk 1:00:00] } g6 { [%clk 1:00:00] } 3. Nc3 { [%clk 1:00:00] } Bg7 { [%clk 1:00:00] } 4. e4 { [%clk 1:00:00] } d6 { [%clk 1:00:00] } 5. f3 { [%clk 1:00:00] } O-O { [%clk 1:00:00] } 6. Be3 { [%clk 1:00:00] } e5 { [%clk 1:00:00] } 7. d5 { [%clk 1:00:00] } Nh5 { [%clk 1:00:00] } 8. Qd2 { [%clk 1:00:00] } Qh4+ { [%clk 1:00:00] } 9. g3 { [%clk 1:00:00] } Qe7 { [%clk 1:00:00] } 10. Nh3 { [%clk 1:00:00] } f5 { [%clk 0:56:44] } 11. exf5 { [%clk 0:58:18] } gxf5 { [%clk 0:55:20] } 12. O-O-O { [%clk 0:57:22] } Na6 { [%clk 0:52:30] } 13. Re1 { [%clk 0:52:22] } Nf6 { [%clk 0:48:20] } 14. Ng5 { [%clk 0:50:43] } c6 { [%clk 0:47:38] } 15. h4 { [%clk 0:50:01] } h6 { [%clk 0:46:10] } 16. Nh3 { [%clk 0:49:18] } cxd5 { [%clk 0:45:06] } 17. Bxh6 { [%clk 0:47:13] } Bxh6 { [%clk 0:44:17] } 18. Qxh6 { [%clk 0:45:59] } Bd7 { [%clk 0:43:34] } 19. cxd5 { [%clk 0:45:15] } Nc5 { [%clk 0:42:50] } 20. Kb1 { [%clk 0:44:14] } Qg7 { [%clk 0:41:29] } 21. Qd2 { [%clk 0:42:39] } e4 { [%clk 0:40:55] } 22. b4 { [%clk 0:40:31] } Na4 { [%clk 0:39:58] } 23. Nxa4 { [%clk 0:39:13] } Bxa4 { [%clk 0:38:39] } 24. Ng5 { [%clk 0:37:47] } Rfc8 { [%clk 0:37:14] } 25. Ne6 { [%clk 0:36:01] } Rc2 { [%clk 0:36:38] } *
|
|
''';
|
|
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
expect(root.mainline.length, equals(48));
|
|
|
|
final root2 = Root.fromPgnGame(PgnGame.parsePgn(pgn2));
|
|
expect(root2.mainline.length, equals(50));
|
|
|
|
root2.merge(root);
|
|
expect(root2.mainline.length, equals(50));
|
|
expect(root2.mainline.last.sanMove.san, equals('Rc2'));
|
|
|
|
for (final nodes in IterableZip([root.mainline, root2.mainline])) {
|
|
final [node1, node2] = nodes;
|
|
expect(node1.sanMove, equals(node2.sanMove));
|
|
expect(node1.position.fen, equals(node2.position.fen));
|
|
expect(node1.clock, equals(node2.clock));
|
|
}
|
|
});
|
|
|
|
test('preserve variations', () {
|
|
const pgn = '''
|
|
1. d4 { [%clk 1:00:00] } Nf6 { [%clk 1:00:00] } 2. c4 { [%clk 1:00:00] } g6 { [%clk 1:00:00] } 3. Nc3 { [%clk 1:00:00] } Bg7 { [%clk 1:00:00] } 4. e4 { [%clk 1:00:00] } d6 { [%clk 1:00:00] } 5. f3 { [%clk 1:00:00] } O-O { [%clk 1:00:00] } 6. Be3 { [%clk 1:00:00] } e5 { [%clk 1:00:00] } 7. d5 { [%clk 1:00:00] } Nh5 { [%clk 1:00:00] } 8. Qd2 { [%clk 1:00:00] } Qh4+ { [%clk 1:00:00] } 9. g3 { [%clk 1:00:00] } Qe7 { [%clk 1:00:00] } 10. Nh3 { [%clk 1:00:00] } f5 { [%clk 0:56:44] } 11. exf5 { [%clk 0:58:18] } gxf5 { [%clk 0:55:20] } 12. O-O-O { [%clk 0:57:22] } Na6 { [%clk 0:52:30] } 13. Re1 { [%clk 0:52:22] } Nf6 { [%clk 0:48:20] } 14. Ng5 { [%clk 0:50:43] } c6 { [%clk 0:47:38] } 15. h4 { [%clk 0:50:01] } h6 { [%clk 0:46:10] } 16. Nh3 { [%clk 0:49:18] } cxd5 { [%clk 0:45:06] } 17. Bxh6 { [%clk 0:47:13] } Bxh6 { [%clk 0:44:17] } 18. Qxh6 { [%clk 0:45:59] } Bd7 { [%clk 0:43:34] } 19. cxd5 { [%clk 0:45:15] } Nc5 { [%clk 0:42:50] } 20. Kb1 { [%clk 0:44:14] } Qg7 { [%clk 0:41:29] } 21. Qd2 { [%clk 0:42:39] } e4 { [%clk 0:40:55] } 22. b4 { [%clk 0:40:31] } Na4 { [%clk 0:39:58] } 23. Nxa4 { [%clk 0:39:13] } Bxa4 { [%clk 0:38:39] } 24. Ng5 { [%clk 0:37:47] } Rfc8 { [%clk 0:37:14] } 25. Ne6 { [%clk 0:36:01] } Rc2 { [%clk 0:36:38] } 26. Qe3 { [%clk 0:34:49] } Nxd5 { [%clk 0:34:34] } ( 26... Qe7 27. Rc1 ) *
|
|
''';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
expect(root.mainline.length, equals(52));
|
|
|
|
const pgn2 = '''
|
|
1. d4 { [%clk 1:00:00] } Nf6 { [%clk 1:00:00] } 2. c4 { [%clk 1:00:00] } g6 { [%clk 1:00:00] } 3. Nc3 { [%clk 1:00:00] } Bg7 { [%clk 1:00:00] } 4. e4 { [%clk 1:00:00] } d6 { [%clk 1:00:00] } 5. f3 { [%clk 1:00:00] } O-O { [%clk 1:00:00] } 6. Be3 { [%clk 1:00:00] } e5 { [%clk 1:00:00] } 7. d5 { [%clk 1:00:00] } Nh5 { [%clk 1:00:00] } 8. Qd2 { [%clk 1:00:00] } Qh4+ { [%clk 1:00:00] } 9. g3 { [%clk 1:00:00] } Qe7 { [%clk 1:00:00] } 10. Nh3 { [%clk 1:00:00] } f5 { [%clk 0:56:44] } 11. exf5 { [%clk 0:58:18] } gxf5 { [%clk 0:55:20] } 12. O-O-O { [%clk 0:57:22] } Na6 { [%clk 0:52:30] } 13. Re1 { [%clk 0:52:22] } Nf6 { [%clk 0:48:20] } 14. Ng5 { [%clk 0:50:43] } c6 { [%clk 0:47:38] } 15. h4 { [%clk 0:50:01] } h6 { [%clk 0:46:10] } 16. Nh3 { [%clk 0:49:18] } cxd5 { [%clk 0:45:06] } 17. Bxh6 { [%clk 0:47:13] } Bxh6 { [%clk 0:44:17] } 18. Qxh6 { [%clk 0:45:59] } Bd7 { [%clk 0:43:34] } 19. cxd5 { [%clk 0:45:15] } Nc5 { [%clk 0:42:50] } 20. Kb1 { [%clk 0:44:14] } Qg7 { [%clk 0:41:29] } 21. Qd2 { [%clk 0:42:39] } e4 { [%clk 0:40:55] } 22. b4 { [%clk 0:40:31] } Na4 { [%clk 0:39:58] } 23. Nxa4 { [%clk 0:39:13] } Bxa4 { [%clk 0:38:39] } 24. Ng5 { [%clk 0:37:47] } Rfc8 { [%clk 0:37:14] } 25. Ne6 { [%clk 0:36:01] } Rc2 { [%clk 0:36:38] } 26. Qe3 { [%clk 0:34:49] } Nxd5 { [%clk 0:34:34] } 27. Nxg7 { [%clk 0:34:17] } Nxe3 { [%clk 0:34:04] } 28. Rxe3 { [%clk 0:33:12] } Kxg7 { [%clk 0:33:33] } 29. Ra3 { [%clk 0:31:18] } Rac8 { [%clk 0:32:46] } 30. Bh3 { [%clk 0:30:15] } *
|
|
''';
|
|
final root2 = Root.fromPgnGame(PgnGame.parsePgn(pgn2));
|
|
expect(root2.mainline.length, equals(59));
|
|
|
|
root2.merge(root);
|
|
expect(root2.mainline.length, equals(59));
|
|
expect(root2.makePgn(), '''
|
|
1. d4 { [%clk 1:00:00] } Nf6 { [%clk 1:00:00] } 2. c4 { [%clk 1:00:00] } g6 { [%clk 1:00:00] } 3. Nc3 { [%clk 1:00:00] } Bg7 { [%clk 1:00:00] } 4. e4 { [%clk 1:00:00] } d6 { [%clk 1:00:00] } 5. f3 { [%clk 1:00:00] } O-O { [%clk 1:00:00] } 6. Be3 { [%clk 1:00:00] } e5 { [%clk 1:00:00] } 7. d5 { [%clk 1:00:00] } Nh5 { [%clk 1:00:00] } 8. Qd2 { [%clk 1:00:00] } Qh4+ { [%clk 1:00:00] } 9. g3 { [%clk 1:00:00] } Qe7 { [%clk 1:00:00] } 10. Nh3 { [%clk 1:00:00] } f5 { [%clk 0:56:44] } 11. exf5 { [%clk 0:58:18] } gxf5 { [%clk 0:55:20] } 12. O-O-O { [%clk 0:57:22] } Na6 { [%clk 0:52:30] } 13. Re1 { [%clk 0:52:22] } Nf6 { [%clk 0:48:20] } 14. Ng5 { [%clk 0:50:43] } c6 { [%clk 0:47:38] } 15. h4 { [%clk 0:50:01] } h6 { [%clk 0:46:10] } 16. Nh3 { [%clk 0:49:18] } cxd5 { [%clk 0:45:06] } 17. Bxh6 { [%clk 0:47:13] } Bxh6 { [%clk 0:44:17] } 18. Qxh6 { [%clk 0:45:59] } Bd7 { [%clk 0:43:34] } 19. cxd5 { [%clk 0:45:15] } Nc5 { [%clk 0:42:50] } 20. Kb1 { [%clk 0:44:14] } Qg7 { [%clk 0:41:29] } 21. Qd2 { [%clk 0:42:39] } e4 { [%clk 0:40:55] } 22. b4 { [%clk 0:40:31] } Na4 { [%clk 0:39:58] } 23. Nxa4 { [%clk 0:39:13] } Bxa4 { [%clk 0:38:39] } 24. Ng5 { [%clk 0:37:47] } Rfc8 { [%clk 0:37:14] } 25. Ne6 { [%clk 0:36:01] } Rc2 { [%clk 0:36:38] } 26. Qe3 { [%clk 0:34:49] } Nxd5 { [%clk 0:34:34] } ( 26... Qe7 27. Rc1 ) 27. Nxg7 { [%clk 0:34:17] } Nxe3 { [%clk 0:34:04] } 28. Rxe3 { [%clk 0:33:12] } Kxg7 { [%clk 0:33:33] } 29. Ra3 { [%clk 0:31:18] } Rac8 { [%clk 0:32:46] } 30. Bh3 { [%clk 0:30:15] } *
|
|
''');
|
|
});
|
|
|
|
test('preserve evals', () {
|
|
const pgn = '''
|
|
1. d4 { [%clk 1:00:00] } Nf6 { [%clk 1:00:00] } 2. c4 { [%clk 1:00:00] } g6 { [%clk 1:00:00] } 3. Nc3 { [%clk 1:00:00] } Bg7 { [%clk 1:00:00] } 4. e4 { [%clk 1:00:00] } d6 { [%clk 1:00:00] } 5. f3 { [%clk 1:00:00] } O-O { [%clk 1:00:00] } 6. Be3 { [%clk 1:00:00] } e5 { [%clk 1:00:00] } 7. d5 { [%clk 1:00:00] } Nh5 { [%clk 1:00:00] } 8. Qd2 { [%clk 1:00:00] } Qh4+ { [%clk 1:00:00] } 9. g3 { [%clk 1:00:00] } Qe7 { [%clk 1:00:00] } 10. Nh3 { [%clk 1:00:00] } f5 { [%clk 0:56:44] } 11. exf5 { [%clk 0:58:18] } gxf5 { [%clk 0:55:20] } 12. O-O-O { [%clk 0:57:22] } Na6 { [%clk 0:52:30] } 13. Re1 { [%clk 0:52:22] } Nf6 { [%clk 0:48:20] } 14. Ng5 { [%clk 0:50:43] } c6 { [%clk 0:47:38] } 15. h4 { [%clk 0:50:01] } h6 { [%clk 0:46:10] } 16. Nh3 { [%clk 0:49:18] } cxd5 { [%clk 0:45:06] } 17. Bxh6 { [%clk 0:47:13] } Bxh6 { [%clk 0:44:17] } 18. Qxh6 { [%clk 0:45:59] } Bd7 { [%clk 0:43:34] } 19. cxd5 { [%clk 0:45:15] } Nc5 { [%clk 0:42:50] } 20. Kb1 { [%clk 0:44:14] } Qg7 { [%clk 0:41:29] } 21. Qd2 { [%clk 0:42:39] } e4 { [%clk 0:40:55] } 22. b4 { [%clk 0:40:31] } Na4 { [%clk 0:39:58] } 23. Nxa4 { [%clk 0:39:13] } Bxa4 { [%clk 0:38:39] } 24. Ng5 { [%clk 0:37:47] } Rfc8 { [%clk 0:37:14] } 25. Ne6 { [%clk 0:36:01] } Rc2 { [%clk 0:36:38] } 26. Qe3 { [%clk 0:34:49] } Nxd5 { [%clk 0:34:34] } 27. Nxg7 { [%clk 0:34:17] } Nxe3 { [%clk 0:34:04] } 28. Rxe3 { [%clk 0:33:12] } Kxg7 { [%clk 0:33:33] } 29. Ra3 { [%clk 0:31:18] } Rac8 { [%clk 0:32:46] } 30. Bh3 { [%clk 0:30:15] } *
|
|
''';
|
|
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
expect(root.mainline.length, equals(59));
|
|
final localEval = LocalEval(
|
|
position: Chess.initial,
|
|
depth: 22,
|
|
nodes: 100000,
|
|
pvs: IList<PvData>(),
|
|
millis: 1230900,
|
|
searchTime: const Duration(milliseconds: 1230900),
|
|
cp: 23,
|
|
threatMode: false,
|
|
);
|
|
root.mainline.last.eval = localEval;
|
|
|
|
const pgn2 = '''
|
|
1. d4 { [%clk 1:00:00] } Nf6 { [%clk 1:00:00] } 2. c4 { [%clk 1:00:00] } g6 { [%clk 1:00:00] } 3. Nc3 { [%clk 1:00:00] } Bg7 { [%clk 1:00:00] } 4. e4 { [%clk 1:00:00] } d6 { [%clk 1:00:00] } 5. f3 { [%clk 1:00:00] } O-O { [%clk 1:00:00] } 6. Be3 { [%clk 1:00:00] } e5 { [%clk 1:00:00] } 7. d5 { [%clk 1:00:00] } Nh5 { [%clk 1:00:00] } 8. Qd2 { [%clk 1:00:00] } Qh4+ { [%clk 1:00:00] } 9. g3 { [%clk 1:00:00] } Qe7 { [%clk 1:00:00] } 10. Nh3 { [%clk 1:00:00] } f5 { [%clk 0:56:44] } 11. exf5 { [%clk 0:58:18] } gxf5 { [%clk 0:55:20] } 12. O-O-O { [%clk 0:57:22] } Na6 { [%clk 0:52:30] } 13. Re1 { [%clk 0:52:22] } Nf6 { [%clk 0:48:20] } 14. Ng5 { [%clk 0:50:43] } c6 { [%clk 0:47:38] } 15. h4 { [%clk 0:50:01] } h6 { [%clk 0:46:10] } 16. Nh3 { [%clk 0:49:18] } cxd5 { [%clk 0:45:06] } 17. Bxh6 { [%clk 0:47:13] } Bxh6 { [%clk 0:44:17] } 18. Qxh6 { [%clk 0:45:59] } Bd7 { [%clk 0:43:34] } 19. cxd5 { [%clk 0:45:15] } Nc5 { [%clk 0:42:50] } 20. Kb1 { [%clk 0:44:14] } Qg7 { [%clk 0:41:29] } 21. Qd2 { [%clk 0:42:39] } e4 { [%clk 0:40:55] } 22. b4 { [%clk 0:40:31] } Na4 { [%clk 0:39:58] } 23. Nxa4 { [%clk 0:39:13] } Bxa4 { [%clk 0:38:39] } 24. Ng5 { [%clk 0:37:47] } Rfc8 { [%clk 0:37:14] } 25. Ne6 { [%clk 0:36:01] } Rc2 { [%clk 0:36:38] } 26. Qe3 { [%clk 0:34:49] } Nxd5 { [%clk 0:34:34] } 27. Nxg7 { [%clk 0:34:17] } Nxe3 { [%clk 0:34:04] } 28. Rxe3 { [%clk 0:33:12] } Kxg7 { [%clk 0:33:33] } 29. Ra3 { [%clk 0:31:18] } Rac8 { [%clk 0:32:46] } 30. Bh3 { [%clk 0:30:15] } Bd7 { [%clk 0:32:05] } 31. fxe4 { [%clk 0:29:38] } R8c4 { [%clk 0:31:11] } 32. Rxa7 { [%clk 0:27:46] } Bc6 { [%clk 0:30:37] } 33. Bxf5 { [%clk 0:27:20] } Re2 { [%clk 0:29:32] } 34. b5 { [%clk 0:26:56] } Rb4+ { [%clk 0:29:02] } 35. Ka1 { [%clk 0:25:59] } Rxb5 { [%clk 0:28:13] } 36. Rb1 { [%clk 0:25:17] } Rc5 { [%clk 0:27:47] } 37. h5 { [%clk 0:23:42] } Rh2 { [%clk 0:27:22] } 38. g4 { [%clk 0:22:55] } Kf6 { [%clk 0:26:59] } 39. Ra3 { [%clk 0:22:10] } Rc4 { [%clk 0:26:36] } 40. Re1 { [%eval 1.17,33] [%clk 0:19:30] } *
|
|
''';
|
|
|
|
final root2 = Root.fromPgnGame(PgnGame.parsePgn(pgn2));
|
|
expect(root2.mainline.length, equals(79));
|
|
|
|
root2.merge(root);
|
|
expect(root2.mainline.length, equals(79));
|
|
|
|
for (final nodes in IterableZip([root.mainline, root2.mainline])) {
|
|
final [node1, node2] = nodes;
|
|
expect(node1.sanMove, equals(node2.sanMove));
|
|
expect(node1.position.fen, equals(node2.position.fen));
|
|
expect(node1.clock, equals(node2.clock));
|
|
}
|
|
// one new external eval
|
|
expect(root2.mainline.where((n) => n.externalEval != null).length, equals(1));
|
|
// one old local eval preseved
|
|
expect(root2.mainline.where((node) => node.eval != null).length, equals(1));
|
|
expect(
|
|
root2.mainline.firstWhereOrNull((node) => node.eval != null)?.eval,
|
|
equals(localEval),
|
|
);
|
|
});
|
|
});
|
|
|
|
group('convert alternative castling move', () {
|
|
void makeTestAltCastlingMove(String pgn, String alt1, String alt2) {
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
final initialPath = root.mainlinePath;
|
|
final initialPng = root.makePgn();
|
|
|
|
final move = Move.parse(alt1);
|
|
expect(move, isNotNull);
|
|
|
|
final newMove = root.convertAltCastlingMove(move!);
|
|
expect(newMove, isNotNull);
|
|
expect(newMove, Move.parse(alt2));
|
|
expect(root.mainline.last.sanMove.move, newMove);
|
|
|
|
final previousUciPath = root.mainlinePath.penultimate;
|
|
final (newPath, isNewNode) = root.addMoveAt(previousUciPath, move);
|
|
expect(newPath, initialPath);
|
|
expect(isNewNode, isFalse);
|
|
expect(root.makePgn(), initialPng);
|
|
}
|
|
|
|
test('e1g1 -> e1h1', () {
|
|
makeTestAltCastlingMove('1. e4 e5 2. Nf3 Nf6 3. Bc4 Bc5 4. O-O', 'e1g1', 'e1h1');
|
|
});
|
|
|
|
test('e8g8 -> e8h8', () {
|
|
makeTestAltCastlingMove('1. e4 e5 2. Nf3 Nf6 3. Bc4 Bc5 4. O-O O-O', 'e8g8', 'e8h8');
|
|
});
|
|
|
|
test('e1c1 -> e1a1', () {
|
|
makeTestAltCastlingMove(
|
|
'1. d4 d5 2. Nc3 Nc6 3. Be3 Be6 4. Qd3 Qd6 5. O-O-O',
|
|
'e1c1',
|
|
'e1a1',
|
|
);
|
|
});
|
|
|
|
test('e8c8 -> e8a8', () {
|
|
makeTestAltCastlingMove(
|
|
'1. d4 d5 2. Nc3 Nc6 3. Be3 Be6 4. Qd3 Qd6 5. O-O-O O-O-O',
|
|
'e8c8',
|
|
'e8a8',
|
|
);
|
|
});
|
|
test('only convert king moves in altCastlingMove', () {
|
|
const pgn = '1. e4 e5 2. Bc4 Qh4 3. Nf3 Qxh2 4. Ke2 Qxh1 5. Qe1 Qh5 6. Qh1';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
final initialPng = root.makePgn();
|
|
final previousUciPath = root.mainlinePath.penultimate;
|
|
final move = Move.parse('e1g1');
|
|
root.addMoveAt(previousUciPath, move!);
|
|
expect(root.makePgn(), isNot(initialPng));
|
|
});
|
|
|
|
test('do not convert castling move if rook is on the alternative castling square', () {
|
|
const pgn = '[FEN "rnbqkbnr/pppppppp/8/8/8/2NBQ3/PPPPPPPP/2R1KBNR w KQkq - 0 1"]';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
final initialPng = root.makePgn();
|
|
final previousUciPath = root.mainlinePath.penultimate;
|
|
final move = Move.parse('e1c1');
|
|
root.addMoveAt(previousUciPath, move!);
|
|
expect(root.makePgn(), isNot(initialPng));
|
|
expect(root.mainline.last.sanMove.move, move);
|
|
});
|
|
});
|
|
});
|
|
|
|
group('ViewNode', () {
|
|
test('mainline', () {
|
|
const pgn = '1. e4 e5 (1... d5 2. a4) 2. a4';
|
|
final root = Root.fromPgnGame(PgnGame.parsePgn(pgn));
|
|
final viewRoot = root.view;
|
|
|
|
{
|
|
final mainline = viewRoot.mainline;
|
|
|
|
expect(mainline.length, equals(3));
|
|
final list = mainline.toList();
|
|
expect(list[0].sanMove, equals(SanMove('e4', Move.parse('e2e4')!)));
|
|
expect(list[1].sanMove, equals(SanMove('e5', Move.parse('e7e5')!)));
|
|
expect(list[2].sanMove, equals(SanMove('a4', Move.parse('a2a4')!)));
|
|
}
|
|
|
|
{
|
|
final childMainline = viewRoot.children.first.mainline;
|
|
|
|
expect(childMainline.length, equals(2));
|
|
final list = childMainline.toList();
|
|
expect(list[0].sanMove, equals(SanMove('e5', Move.parse('e7e5')!)));
|
|
expect(list[1].sanMove, equals(SanMove('a4', Move.parse('a2a4')!)));
|
|
}
|
|
});
|
|
});
|
|
}
|
|
|
|
const fisherSpasskyPgn = '''
|
|
[Event "F/S Return Match"]
|
|
[Site "Belgrade, Serbia Yugoslavia|JUG"]
|
|
[Date "1992.11.04"]
|
|
[Round "29"]
|
|
[White "Fischer, Robert J."]
|
|
[Black "Spassky, Boris V."]
|
|
[Result "1/2-1/2"]
|
|
|
|
1. e4 e5 2. Nf3 Nc6 3. Bb5 {This opening is called the Ruy Lopez.} 3... a6
|
|
4. Ba4 Nf6 5. O-O Be7 6. Re1 b5 7. Bb3 d6 8. c3 O-O 9. h3 Nb8 10. d4 Nbd7
|
|
11. c4 c6 12. cxb5 axb5 13. Nc3 Bb7 14. Bg5 b4 15. Nb1 h6 16. Bh4 c5 17. dxe5
|
|
Nxe4 18. Bxe7 Qxe7 19. exd6 Qf6 20. Nbd2 Nxd6 21. Nc4 Nxc4 22. Bxc4 Nb6
|
|
23. Ne5 Rae8 24. Bxf7+ Rxf7 25. Nxf7 Rxe1+ 26. Qxe1 Kxf7 27. Qe3 Qg5 28. Qxg5
|
|
hxg5 29. b3 Ke6 30. a3 Kd6 31. axb4 cxb4 32. Ra5 Nd5 33. f3 Bc8 34. Kf2 Bf5
|
|
35. Ra7 g6 36. Ra6+ Kc5 37. Ke1 Nf4 38. g3 Nxh3 39. Kd2 Kb5 40. Rd6 Kc5 41. Ra6
|
|
Nf2 42. g4 Bd3 43. Re6 1/2-1/2''';
|