Module:Pgn/testcases

-- Unit tests for Module:Pgn. Click talk page to run tests. local p = require('Module:UnitTests') local pgnModule = require('Module:Pgn') -- test extraction of moves to FEN notation - main functionality of the module function p:testFullPgnMoves local pgn=[[[Event "World Cup"] [Site "Khanty-Mansiysk"] [Date "2007.11.25"] [Round "1.2"] [White "Mamedyarov, Shakhriyar"] [Black "Abdel Razik, Khaled"] [Result "1-0"] [ECO "D43"] [WhiteElo "2752"] [BlackElo "2389"] [PlyCount "65"] [EventDate "2007.11.24"] [EventType "k.o."] [EventRounds "7"] [EventCountry "RUS"] [Source "ChessBase"] [SourceDate "2008.01.30"]

1. d4 e6 2. c4 Nf6 3. Nf3 d5 4. Nc3 c6 5. Bg5 h6 6. Bxf6 Qxf6 7. Qc2 Nd7 8. e4 dxe4 9. Qxe4 Bb4 10. Bd3 g6 11. O-O O-O 12. Qe3 Bxc3 13. bxc3 Kg7 14. a4 b6 15. a5 Ba6 16. Rfb1 Rfc8 17. h4 b5 18. cxb5 Bxb5 19. Bxb5 cxb5 20. Rxb5 Rab8 21. Rab1 Qf5 22. Rxb8 Rxb8 23. Rxb8 Nxb8 24. Ne5 h5 25. c4 f6 26. Qb3 fxe5 27. Qxb8 exd4 28. Qxa7+ Kg8 29. Qb8+ Kh7 30. a6 d3 31. a7 d2 32. a8=Q d1=Q+ 33. Kh2 1-0 ]]

metadata, moves = pgnModule.main(pgn) self:preprocess_equals(   	moves[1],    	'rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR'    ) self:preprocess_equals(   	moves[#moves],    	'QQ6/7k/4p1p1/5q1p/2P4P/8/5PPK/3q4'    ) end

--tests metadata extraction function p:testFullPgnMetadata local pgn=[[[Event "World Cup"] [Site "Khanty-Mansiysk"] [Date "2007.11.25"] [Round "1.2"] [White "Mamedyarov, Shakhriyar"] [Black "Abdel Razik, Khaled"] [Result "1-0"] [ECO "D43"] [WhiteElo "2752"] [BlackElo "2389"] [PlyCount "65"] [EventDate "2007.11.24"] [EventType "k.o."] [EventRounds "7"] [EventCountry "RUS"] [Source "ChessBase"] [SourceDate "2008.01.30"]

1. d4 e6 2. c4 Nf6 3. Nf3 d5 4. Nc3 c6 5. Bg5 h6 6. Bxf6 Qxf6 7. Qc2 Nd7 8. e4 dxe4 9. Qxe4 Bb4 10. Bd3 g6 11. O-O O-O 12. Qe3 Bxc3 13. bxc3 Kg7 14. a4 b6 15. a5 Ba6 16. Rfb1 Rfc8 17. h4 b5 18. cxb5 Bxb5 19. Bxb5 cxb5 20. Rxb5 Rab8 21. Rab1 Qf5 22. Rxb8 Rxb8 23. Rxb8 Nxb8 24. Ne5 h5 25. c4 f6 26. Qb3 fxe5 27. Qxb8 exd4 28. Qxa7+ Kg8 29. Qb8+ Kh7 30. a6 d3 31. a7 d2 32. a8=Q d1=Q+ 33. Kh2 1-0

]]	metadata, moves = pgnModule.main(pgn) self:preprocess_equals( metadata['Black'] or '', 'Abdel Razik, Khaled' ) self:preprocess_equals( metadata['White'] or '', 'Mamedyarov, Shakhriyar' ) end

-- tests algebric notation PGN with no metadata function p:testAlgebricNotation local pgn=1. d4 e6 2. c4 Nf6 3. Nf3 d5 4. Nc3 c6 5. Bg5 h6 6. Bxf6 Qxf6 7. Qc2 Nd7 8. e4 dxe4 9. Qxe4 Bb4 10. Bd3 g6 11. O-O O-O 12. Qe3 Bxc3 13. bxc3 Kg7 14. a4 b6 15. a5 Ba6 16. Rfb1 Rfc8 17. h4 b5 18. cxb5 Bxb5 19. Bxb5 cxb5 20. Rxb5 Rab8 21. Rab1 Qf5 22. Rxb8 Rxb8 23. Rxb8 Nxb8 24. Ne5 h5 25. c4 f6 26. Qb3 fxe5 27. Qxb8 exd4 28. Qxa7+ Kg8 29. Qb8+ Kh7 30. a6 d3 31. a7 d2 32. a8=Q d1=Q+ 33. Kh2 1-0 local metadata, moves success, metadata, moves = pcall(pgnModule.main,pgn) if not success then moves = {'fail'} end self:preprocess_equals(   	moves[1],    	'rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR'    ) self:preprocess_equals(   	moves[#moves],    	'QQ6/7k/4p1p1/5q1p/2P4P/8/5PPK/3q4'    ) end

return p