Update TetrisEngineTest.cs

Author: cdwcgt

osu.Game.Tests/NonVisual/Tetris/TetrisEngineTest.cs

@@ -8,6 +8,7 @@
 using NUnit.Framework;
 using osu.Game.Screens.Tetris;
 using osu.Game.Screens.Tetris.Definitions;
+using osu.Game.Screens.Tetris.Judgements;
 using osu.Game.Screens.Tetris.Rotations;
 using osuTK.Graphics;
 
@@ -331,6 +332,114 @@ public void TestGravityPropertiesStayInSync()
             Assert.That(settings.BaseGravityInterval, Is.EqualTo(1000.0 / (0.02 * 60)).Within(0.000001));
         }
 
+        [Test]
+        public void TestComboTracksConsecutiveLineClears()
+        {
+            var pieceSet = createSingleCellSequencePieceSet();
+            var engine = new TetrisEngine(new TetrisGameSettings
+            {
+                PieceSet = pieceSet,
+                RotationSystem = pieceSet.DefaultRotationSystem,
+                BoardWidth = 2,
+                VisibleRows = 8,
+                HiddenRows = 2,
+                RandomSeed = 5,
+            });
+
+            engine.Press(TetrisAction.HardDrop);
+            Assert.That(engine.Combo, Is.EqualTo(0));
+
+            engine.Press(TetrisAction.MoveRight);
+            engine.Press(TetrisAction.HardDrop);
+            Assert.That(engine.Combo, Is.EqualTo(1));
+
+            engine.Press(TetrisAction.HardDrop);
+            Assert.That(engine.Combo, Is.EqualTo(0));
+        }
+
+        [Test]
+        public void TestBackToBackTracksTetrisesAndBreaksOnRegularLineClear()
+        {
+            var pieceSet = createBackToBackTestPieceSet();
+            var engine = new TetrisEngine(new TetrisGameSettings
+            {
+                PieceSet = pieceSet,
+                RotationSystem = pieceSet.DefaultRotationSystem,
+                BoardWidth = 1,
+                VisibleRows = 12,
+                HiddenRows = 2,
+                RandomSeed = 6,
+            });
+
+            engine.Press(TetrisAction.HardDrop);
+            Assert.That(engine.BackToBack, Is.True);
+            Assert.That(engine.Combo, Is.EqualTo(1));
+
+            engine.Press(TetrisAction.HardDrop);
+            Assert.That(engine.BackToBack, Is.False);
+            Assert.That(engine.Combo, Is.EqualTo(2));
+
+            engine.Press(TetrisAction.HardDrop);
+            Assert.That(engine.BackToBack, Is.True);
+            Assert.That(engine.Combo, Is.EqualTo(3));
+        }
+
+        [Test]
+        public void TestCustomLockJudgementSystemCanDefineBackToBack()
+        {
+            var pieceSet = createSingleCellSequencePieceSet();
+            var engine = new TetrisEngine(new TetrisGameSettings
+            {
+                PieceSet = pieceSet,
+                RotationSystem = pieceSet.DefaultRotationSystem,
+                BoardWidth = 1,
+                VisibleRows = 8,
+                HiddenRows = 2,
+                RandomSeed = 8,
+                LockJudgementSystem = new TetrisLockJudgementSystem("All Singles", context =>
+                    new TetrisLockJudgement(TetrisSpinKind.Full, context.LinesCleared > 0)),
+            });
+
+            engine.Press(TetrisAction.HardDrop);
+
+            Assert.That(engine.BackToBack, Is.True);
+            Assert.That(engine.LastSpin, Is.EqualTo(TetrisSpinKind.Full));
+        }
+
+        [Test]
+        public void TestStandardJudgementRecognisesFullTSpin()
+        {
+            TetrisPieceDefinition t = TetrisPieceSetDefinitions.Modern.Pieces.Single(p => p.Name == "T");
+            var board = new TetrisPieceDefinition[10, 24];
+
+            board[4, 5] = t;
+            board[6, 5] = t;
+            board[4, 7] = t;
+
+            var context = new TetrisLockContext(t, 1, 4, 5, 2, true, 1, true, new[] { 5, 6 }, board);
+            TetrisLockJudgement judgement = TetrisLockJudgementSystems.Standard.Evaluate(context);
+
+            Assert.That(judgement.Spin, Is.EqualTo(TetrisSpinKind.Full));
+            Assert.That(judgement.MaintainsBackToBack, Is.True);
+        }
+
+        [Test]
+        public void TestAllSpinJudgementRecognisesGenericSpin()
+        {
+            TetrisPieceDefinition l = TetrisPieceSetDefinitions.Modern.Pieces.Single(p => p.Name == "L");
+            var board = new TetrisPieceDefinition[10, 24];
+
+            board[4, 5] = l;
+            board[6, 5] = l;
+            board[5, 4] = l;
+
+            var context = new TetrisLockContext(l, 1, 4, 5, 1, true, 1, true, new[] { 5 }, board);
+            TetrisLockJudgement judgement = TetrisLockJudgementSystems.AllSpin.Evaluate(context);
+
+            Assert.That(judgement.Spin, Is.EqualTo(TetrisSpinKind.Full));
+            Assert.That(judgement.MaintainsBackToBack, Is.True);
+        }
+
         private static TetrisEngine createDasCutEngine(TetrisPieceSetDefinition pieceSet)
         {
             return new TetrisEngine(new TetrisGameSettings
@@ -380,5 +489,48 @@ private static TetrisPieceSetDefinition createRotatingBarPieceSet()
 
             return new TetrisPieceSetDefinition("Rotating Bar", new[] { bar }, new[] { rotationSystem }, rotationSystem);
         }
+
+        private static TetrisPieceSetDefinition createSingleCellSequencePieceSet()
+        {
+            var dot = new TetrisPieceDefinition("Dot", Color4.White, new[]
+            {
+                new[] { new PiecePoint(0, 0) },
+            }, "dot");
+
+            var rotationSystem = new TetrisRotationSystem("Identity", (_, _, _, _) => new[]
+            {
+                new PiecePoint(0, 0),
+            });
+
+            return new TetrisPieceSetDefinition("Single Cell Sequence", new[] { dot }, new[] { rotationSystem }, rotationSystem,
+                (_, pieces) => Enumerable.Repeat(pieces[0], 16));
+        }
+
+        private static TetrisPieceSetDefinition createBackToBackTestPieceSet()
+        {
+            var tetris = new TetrisPieceDefinition("Tetris", Color4.Cyan, new[]
+            {
+                new[]
+                {
+                    new PiecePoint(0, 0),
+                    new PiecePoint(0, 1),
+                    new PiecePoint(0, 2),
+                    new PiecePoint(0, 3),
+                },
+            }, "tetris");
+
+            var dot = new TetrisPieceDefinition("Dot", Color4.White, new[]
+            {
+                new[] { new PiecePoint(0, 0) },
+            }, "dot");
+
+            var rotationSystem = new TetrisRotationSystem("Identity", (_, _, _, _) => new[]
+            {
+                new PiecePoint(0, 0),
+            });
+
+            return new TetrisPieceSetDefinition("Back To Back Test", new[] { tetris, dot }, new[] { rotationSystem }, rotationSystem,
+                (_, __) => new[] { tetris, dot, tetris, dot, tetris, dot });
+        }
     }
 }