public void RemoveFullShipsCorrectly()
{
var ships = new[]
{
new { Type = ShipType.Battleship, Start = new CellPosition(0, 0), Vertical = true },
new { Type = ShipType.Cruiser, Start = new CellPosition(5, 5), Vertical = false },
new { Type = ShipType.Submarine, Start = new CellPosition(9, 9), Vertical = true }
};
foreach (var ship in ships)
{
builder.TryAddFullShip(ship.Type, ship.Start, ship.Vertical);
}
foreach (var ship in ships)
{
builder.TryRemoveFullShip(ship.Type, ship.Start, ship.Vertical).Should().BeTrue();
}
foreach (var position in builder.EnumeratePositions())
{
builder[position].Should().BeFalse();
}
}
protected IGameField GenerateNewPrediction()
{
var builder = new GameFieldBuilder();
foreach (var position in OpponentFieldKnowledge.EnumeratePositions())
{
if (OpponentFieldKnowledge[position] == true)
{
builder.TryAddShipCell(position);
}
}
var generator = new RandomFieldGenerator(builder);
var damagedShip = FindDamagedShip().ToList();
if (damagedShip.Any())
{
foreach (var cell in damagedShip)
{
builder.TryRemoveShipCell(cell);
}
var variants = new[] { 4, 3, 2, 1 }.SelectMany(x => new[]
{
new { Ship = (ShipType)x, Vertical = true },
new { Ship = (ShipType)x, Vertical = false }
}).SelectMany(x => GenerateContinuesForDamagedShip(damagedShip, builder, x.Vertical, x.Ship));
foreach (var variant in variants)
{
builder.TryAddFullShip(variant.Item1, variant.Item2, variant.Item3);
var prediction = generator.Generate(x => OpponentFieldKnowledge[x] != false);
if (prediction != null)
{
return(prediction);
}
builder.TryRemoveFullShip(variant.Item1, variant.Item2, variant.Item3);
}
}
return(generator.Generate(x => OpponentFieldKnowledge[x] != false));
}
