build.gradle

 
 plugins {
+    id("com.github.johnrengelman.shadow") version "8.1.1"
     id 'application'
-    id "org.openjfx.javafxplugin" version "0.0.14"
+    id 'java'
+    id "org.openjfx.javafxplugin" version "0.1.0"
 }
 
 javafx {
-    version = "20"
-    modules = [ 'javafx.controls', 'javafx.fxml' ]
+    version = "21"
+    modules = [ 'javafx.controls', 'javafx.fxml', 'javafx.graphics']
 }
 
 
@@ -15,9 +17,7 @@ repositories {
 }
 
 dependencies {
-    testRuntimeOnly("org.junit.platform:junit-platform-launcher") {
-        because("Only needed to run tests in a version of IntelliJ IDEA that bundles older versions")
-    }
+    testRuntimeOnly("org.junit.platform:junit-platform-launcher")
     testRuntimeOnly("org.junit.jupiter:junit-jupiter-engine")
     testImplementation("org.assertj:assertj-core:3.24.2")
     testImplementation("org.junit.jupiter:junit-jupiter:5.10.0")
@@ -28,5 +28,7 @@ test {
 }
 
 application {
-    mainClass.set("FirefighterApplication")
+    mainClass.set("app.SimulatorMain")
 }
+
+

gradle/wrapper/gradle-wrapper.properties

 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
-distributionUrl=https\://services.gradle.org/distributions/gradle-8.4-bin.zip
+distributionUrl=https\://services.gradle.org/distributions/gradle-8.10.2-bin.zip
 zipStoreBase=GRADLE_USER_HOME
 zipStorePath=wrapper/dists

settings.gradle

 rootProject.name = 'firefighter'
+include 'src:main:aapp'
+findProject(':src:main:aapp')?.name = 'aapp'
 

src/main/java/FirefighterApplication.java → src/main/java/app/SimulatorApplication.java

+package app;
+
 import controller.Controller;
-import javafx.application.Application;
 import javafx.application.Platform;
 import javafx.fxml.FXMLLoader;
 import javafx.scene.Parent;
 import javafx.scene.Scene;
 import javafx.stage.Stage;
-import model.FirefighterBoard;
 
 import java.io.IOException;
 import java.net.URL;
 
-public class FirefighterApplication extends Application {
+public class SimulatorApplication extends javafx.application.Application {
   private static final String VIEW_RESOURCE_PATH = "/view/view.fxml";
   private static final String APP_NAME = "Firefighter simulator";
-  private static final int ROW_COUNT = 50;
-  private static final int COLUMN_COUNT = 50;
-  private static final int SQUARE_WIDTH = 20;
-  private static final int SQUARE_HEIGHT = 20;
+  private static final int ROW_COUNT = 20;
+  private static final int COLUMN_COUNT = 20;
+  private static final int BOX_WIDTH = 50;
+  private static final int BOX_HEIGHT = 50;
   public static final int INITIAL_FIRE_COUNT = 3;
   public static final int INITIAL_FIREFIGHTER_COUNT = 6;
 
@@ -26,7 +26,7 @@ public class FirefighterApplication extends Application {
     this.primaryStage = primaryStage;
     this.primaryStage.setTitle(APP_NAME);
     this.primaryStage.setOnCloseRequest(event -> Platform.exit());
-    this.primaryStage.setResizable(false);
+    this.primaryStage.setResizable(true);
     this.primaryStage.sizeToScene();
   }
 
@@ -39,11 +39,11 @@ public class FirefighterApplication extends Application {
 
   private void initializeView() throws IOException {
     FXMLLoader loader = new FXMLLoader();
-    URL location = FirefighterApplication.class.getResource(VIEW_RESOURCE_PATH);
+    URL location = SimulatorApplication.class.getResource(VIEW_RESOURCE_PATH);
     loader.setLocation(location);
     view = loader.load();
     Controller controller = loader.getController();
-    controller.initialize(SQUARE_WIDTH, SQUARE_HEIGHT, COLUMN_COUNT, ROW_COUNT,
+    controller.initialize(BOX_WIDTH, BOX_HEIGHT, COLUMN_COUNT, ROW_COUNT,
             INITIAL_FIRE_COUNT, INITIAL_FIREFIGHTER_COUNT);
   }
 
@@ -52,4 +52,8 @@ public class FirefighterApplication extends Application {
     primaryStage.setScene(scene);
     primaryStage.show();
   }
+
+  public static void main(String[] args) {
+    launch(args);
+  }
 }

src/main/java/app/SimulatorMain.java

+package app;
+
+public class SimulatorMain {
+  public static void main(String[] args){
+    SimulatorApplication.main(args);
+  }
+}

src/main/java/controller/Controller.java

@@ -7,6 +7,7 @@ import javafx.event.ActionEvent;
 import javafx.event.EventHandler;
 import javafx.fxml.FXML;
 import javafx.scene.control.Button;
+import javafx.scene.control.Label;
 import javafx.scene.control.ToggleButton;
 import javafx.scene.control.ToggleGroup;
 import javafx.util.Duration;
@@ -14,8 +15,8 @@ import javafx.util.Pair;
 import model.Board;
 import model.ModelElement;
 import model.FirefighterBoard;
-import model.Position;
-import view.FirefighterGrid;
+import util.Position;
+import view.Grid;
 import view.ViewElement;
 
 import java.util.ArrayList;
@@ -27,11 +28,17 @@ public class Controller {
 
   public static final int PERIOD_IN_MILLISECONDS = 50;
   @FXML
+  public Button restartButton;
+  @FXML
+  public Button oneStepButton;
+  @FXML
+  public Label generationNumberLabel;
+  @FXML
   private ToggleButton pauseToggleButton;
   @FXML
   private ToggleButton playToggleButton;
   @FXML
-  private FirefighterGrid grid;
+  private Grid<ViewElement> grid;
   private Timeline timeline;
   private Board<List<ModelElement>> board;
 
@@ -48,7 +55,7 @@ public class Controller {
   }
 
   private void setModel(FirefighterBoard firefighterBoard) {
-    this.board = requireNonNull(firefighterBoard, "model is null");
+    this.board = requireNonNull(firefighterBoard, "firefighter.model is null");
   }
 
   private void updateBoard(){
@@ -60,9 +67,10 @@ public class Controller {
       updatedSquares.add(new Pair<>(updatedPosition, viewElement));
     }
     grid.repaint(updatedSquares);
+    updateGenerationLabel(board.stepNumber());
   }
 
-  private void repaintBoard(){
+  private void repaintGrid(){
     int columnCount = board.columnCount();
     int rowCount = board.rowCount();
     ViewElement[][] viewElements = new ViewElement[rowCount][columnCount];
@@ -70,6 +78,7 @@ public class Controller {
       for(int row = 0; row < rowCount; row++)
         viewElements[row][column] = getViewElement(board.getState(new Position(row, column)));
     grid.repaint(viewElements);
+    updateGenerationLabel(board.stepNumber());
   }
 
   private ViewElement getViewElement(List<ModelElement> squareState) {
@@ -99,30 +108,34 @@ public class Controller {
     timeline.pause();
   }
 
-  public void pauseToggleButtonAction(ActionEvent actionEvent) {
+  public void pauseToggleButtonAction() {
     this.pause();
   }
 
-  public void playToggleButtonAction(ActionEvent actionEvent) {
+  public void playToggleButtonAction() {
     this.play();
   }
 
-  public void restartButtonAction(ActionEvent actionEvent) {
+  public void restartButtonAction() {
     this.pause();
     board.reset();
     pauseToggleButton.setSelected(true);
-    repaintBoard();
+    repaintGrid();
   }
 
   public void initialize(int squareWidth, int squareHeight, int columnCount,
                                 int rowCount, int initialFireCount, int initialFirefighterCount) {
-    grid.initialize(squareWidth, squareHeight, columnCount, rowCount);
+    grid.setDimensions(columnCount, rowCount, squareWidth, squareHeight);
     this.setModel(new FirefighterBoard(columnCount, rowCount, initialFireCount, initialFirefighterCount));
-    repaintBoard();
+    repaintGrid();
   }
 
-  public void oneStepButtonAction(ActionEvent actionEvent) {
+  public void oneStepButtonAction() {
     this.pause();
     updateBoard();
   }
+
+  private void updateGenerationLabel(int value){
+    generationNumberLabel.setText(Integer.toString(value));
+  }
 }
\ No newline at end of file

src/main/java/model/Board.java

 package model;
 
+import util.Position;
+
 import java.util.List;
 
+/**
+ * This interface represents a generic board for modeling various state-based systems.
+ *
+ * @param <S> The type of state represented on the board.
+ */
 public interface Board<S> {
+
+  /**
+   * Get the state of the board at a specific position.
+   *
+   * @param position The position on the board for which to retrieve the state.
+   * @return The state at the specified position.
+   */
   S getState(Position position);
+
+  /**
+   * Set the state of a specific position on the board to the specified state.
+   *
+   * @param state The state to set for the given position.
+   * @param position The position on the board for which to set the state.
+   */
+  void setState(S state, Position position);
+
+  /**
+   * Get the number of rows in the board.
+   *
+   * @return The number of rows in the board.
+   */
   int rowCount();
+
+  /**
+   * Get the number of columns in the board.
+   *
+   * @return The number of columns in the board.
+   */
   int columnCount();
+
+  /**
+   * Update the board to its next generation or state. This method may modify the
+   * internal state of the board and return a list of positions that have changed
+   * during the update.
+   *
+   * @return A list of positions that have changed during the update.
+   */
   List<Position> updateToNextGeneration();
+
+  /**
+   * Reset the board to its initial state.
+   */
   void reset();
+
+  /**
+   * Get the current step number or generation of the board.
+   *
+   * @return The current step number or generation.
+   */
+  int stepNumber();
 }
+

src/main/java/model/FirefighterBoard.java

 package model;
 
+import util.Position;
+
 import java.util.*;
 
 
@@ -8,14 +10,30 @@ public class FirefighterBoard implements Board<List<ModelElement>> {
   private final int rowCount;
   private final int initialFireCount;
   private final int initialFirefighterCount;
-  List<Position> firefighterPositions;
-  Set<Position> firePositions;
-  List<Position> firefighterNewPositions;
-  int step = 0;
+  private final TargetStrategy targetStrategy = new TargetStrategy();
+  private List<Position> firefighterPositions;
+  private Set<Position> firePositions;
+  private Map<Position, List<Position>> neighbors = new HashMap();
+  private final Position[][] positions;
+  private int step = 0;
+  private final Random randomGenerator = new Random();
 
   public FirefighterBoard(int columnCount, int rowCount, int initialFireCount, int initialFirefighterCount) {
     this.columnCount = columnCount;
     this.rowCount = rowCount;
+    this.positions = new Position[rowCount][columnCount];
+    for (int column = 0; column < columnCount; column++)
+      for (int row = 0; row < rowCount; row++)
+        positions[row][column] = new Position(row, column);
+    for (int column = 0; column < columnCount; column++)
+      for (int row = 0; row < rowCount; row++) {
+        List<Position> list = new ArrayList<>();
+        if (row > 0) list.add(positions[row - 1][column]);
+        if (column > 0) list.add(positions[row][column - 1]);
+        if (row < rowCount - 1) list.add(positions[row + 1][column]);
+        if (column < columnCount - 1) list.add(positions[row][column + 1]);
+        neighbors.put(positions[row][column], list);
+      }
     this.initialFireCount = initialFireCount;
     this.initialFirefighterCount = initialFirefighterCount;
     initializeElements();
@@ -31,7 +49,7 @@ public class FirefighterBoard implements Board<List<ModelElement>> {
   }
 
   private Position randomPosition() {
-    return new Position((int) (Math.random() * rowCount), (int) (Math.random() * columnCount));
+    return new Position(randomGenerator.nextInt(rowCount), randomGenerator.nextInt(columnCount));
   }
 
   @Override
@@ -56,46 +74,55 @@ public class FirefighterBoard implements Board<List<ModelElement>> {
   }
 
   public List<Position> updateToNextGeneration() {
-    List<Position> result = activateFirefighters();
-    result.addAll(activateFires());
+    List<Position> modifiedPositions = updateFirefighters();
+    modifiedPositions.addAll(updateFires());
     step++;
-    return result;
+    return modifiedPositions;
   }
 
-  private List<Position> activateFires() {
-    List<Position> result = new ArrayList<>();
+  private List<Position> updateFires() {
+    List<Position> modifiedPositions = new ArrayList<>();
     if (step % 2 == 0) {
       List<Position> newFirePositions = new ArrayList<>();
       for (Position fire : firePositions) {
-        newFirePositions.addAll(neighbors(fire));
+        newFirePositions.addAll(neighbors.get(fire));
       }
       firePositions.addAll(newFirePositions);
-      result.addAll(newFirePositions);
+      modifiedPositions.addAll(newFirePositions);
+    }
+    return modifiedPositions;
+
   }
-    return result;
 
+  @Override
+  public int stepNumber() {
+    return step;
   }
 
-  private List<Position> activateFirefighters() {
-    List<Position> result = new ArrayList<>();
-    firefighterNewPositions = new ArrayList<>();
+  private List<Position> updateFirefighters() {
+    List<Position> modifiedPosition = new ArrayList<>();
+    List<Position> firefighterNewPositions = new ArrayList<>();
     for (Position firefighterPosition : firefighterPositions) {
-      Position newFirefighterPosition = neighborClosestToFire(firefighterPosition);
-      result.add(firefighterPosition);
-      result.add(newFirefighterPosition);
+      Position newFirefighterPosition =
+              targetStrategy.neighborClosestToFire(firefighterPosition,
+                      firePositions, neighbors);
       firefighterNewPositions.add(newFirefighterPosition);
       extinguish(newFirefighterPosition);
-      List<Position> neighborFirePositions = neighbors(newFirefighterPosition).stream().filter(firePositions::contains).toList();
+      modifiedPosition.add(firefighterPosition);
+      modifiedPosition.add(newFirefighterPosition);
+      List<Position> neighborFirePositions = neighbors.get(newFirefighterPosition).stream()
+              .filter(firePositions::contains).toList();
       for (Position firePosition : neighborFirePositions)
         extinguish(firePosition);
-      result.addAll(neighborFirePositions);
+      modifiedPosition.addAll(neighborFirePositions);
     }
     firefighterPositions = firefighterNewPositions;
-    return result;
+    return modifiedPosition;
   }
 
   @Override
   public void reset() {
+    step = 0;
     initializeElements();
   }
 
@@ -103,32 +130,18 @@ public class FirefighterBoard implements Board<List<ModelElement>> {
     firePositions.remove(position);
   }
 
-  private List<Position> neighbors(Position position) {
-    List<Position> list = new ArrayList<>();
-    if (position.row() > 0) list.add(new Position(position.row() - 1, position.column()));
-    if (position.column() > 0) list.add(new Position(position.row(), position.column() - 1));
-    if (position.row() < rowCount - 1) list.add(new Position(position.row() + 1, position.column()));
-    if (position.column() < columnCount - 1) list.add(new Position(position.row(), position.column() + 1));
-    return list;
-  }
-
-  private Position neighborClosestToFire(Position position) {
-    Set<Position> seen = new HashSet<>();
-    HashMap<Position, Position> firstMove = new HashMap<>();
-    Queue<Position> toVisit = new LinkedList<>(neighbors(position));
-    for (Position initialMove : toVisit)
-      firstMove.put(initialMove, initialMove);
-    while (!toVisit.isEmpty()) {
-      Position current = toVisit.poll();
-      if (firePositions.contains(current))
-        return firstMove.get(current);
-      for (Position adjacent : neighbors(current)) {
-        if (seen.contains(adjacent)) continue;
-        toVisit.add(adjacent);
-        seen.add(adjacent);
-        firstMove.put(adjacent, firstMove.get(current));
-      }
-    }
-    return position;
+
+  @Override
+  public void setState(List<ModelElement> state, Position position) {
+    firePositions.remove(position);
+    for (; ; ) {
+      if (!firefighterPositions.remove(position)) break;
+    }
+    for (ModelElement element : state) {
+      switch (element) {
+        case FIRE -> firePositions.add(position);
+        case FIREFIGHTER -> firefighterPositions.add(position);
+      }
+    }
   }
 }
\ No newline at end of file

src/main/java/module-info.java

+module firefighter {
+  requires javafx.controls;
+  requires javafx.fxml;
+  requires javafx.graphics;
+  opens controller to javafx.fxml;
+  exports app;
+  opens app to javafx.fxml;
+}

src/main/java/model/Position.java → src/main/java/util/Position.java

-package model;
+package util;
 
 public record Position(int row, int column) {
 

src/main/java/util/TargetStrategy.java

+package model;
+
+import util.Position;
+
+import java.util.*;
+
+public class TargetStrategy {
+
+
+    /**
+     * @param position current position.
+     * @param targets positions that are targeted.
+     * @return the position next to the current position that is on the path to the closest target.
+     */
+    Position neighborClosestToFire(Position position, Collection<Position> targets,
+                                   Map<Position,List<Position>>neighbors) {
+        Set<Position> seen = new HashSet<Position>();
+        HashMap<Position, Position> firstMove = new HashMap<Position, Position>();
+        Queue<Position> toVisit = new LinkedList<Position>(neighbors.get(position));
+        for (Position initialMove : toVisit)
+            firstMove.put(initialMove, initialMove);
+        while (!toVisit.isEmpty()) {
+            Position current = toVisit.poll();
+            if (targets.contains(current))
+                return firstMove.get(current);
+            for (Position adjacent : neighbors.get(current)) {
+                if (seen.contains(adjacent)) continue;
+                toVisit.add(adjacent);
+                seen.add(adjacent);
+                firstMove.put(adjacent, firstMove.get(current));
+            }
+        }
+        return position;
+    }
+}
\ No newline at end of file

src/main/java/view/FirefighterGrid.java

@@ -3,71 +3,97 @@ package view;
 import javafx.scene.canvas.Canvas;
 import javafx.scene.paint.Color;
 import javafx.util.Pair;
-import model.Position;
+import util.Position;
 
 import java.util.List;
 
 public class FirefighterGrid extends Canvas implements Grid<ViewElement>{
 
     private void paintElementAtPosition(ViewElement element, Position position) {
-        paintSquare(position.row(), position.column(), element.color);
+        paintBox(position.row(), position.column(), element.color);
     }
-
-    private int squareWidth;
-    private int squareHeight;
+    private int boxWidth;
+    private int boxHeight;
     private int columnCount;
     private int rowCount;
 
     @Override
     public void repaint(List<Pair<Position, ViewElement>> positionedElements) {
+        clear(positionedElements);
+        paint(positionedElements);
+        paintLines();
+    }
+
+    private void clear(List<Pair<Position, ViewElement>> positionedElements) {
+        for (Pair<Position, ViewElement> positionElement : positionedElements) {
+            Position position = positionElement.getKey();
+            clearBox(position.row(), position.column());
+        }
+    }
+
+    private void paint(List<Pair<Position, ViewElement>> positionedElements) {
         for(Pair<Position, ViewElement> pair : positionedElements){
             paintElementAtPosition(pair.getValue(), pair.getKey());
         }
-        paintLines();
     }
 
     @Override
     public void repaint(ViewElement[][] elements) {
+        clear();
+        paint(elements);
+        paintLines();
+    }
+
+    private void clear() {
+        getGraphicsContext2D().clearRect(0,0,getWidth(), getHeight());
+    }
+
+    private void paint(ViewElement[][] elements) {
         for(int column = 0; column < columnCount; column++)
             for(int row = 0; row < rowCount; row++){
                 paintElementAtPosition(elements[row][column], new Position(row, column));
             }
-        paintLines();
     }
 
-    public int getColumnCount() {
+    public int columnCount() {
         return columnCount;
     }
 
-    public int getRowCount() {
+    public int rowCount() {
         return rowCount;
     }
 
-    public FirefighterGrid(){
-    }
-
-    public void initialize(int squareWidth,
-                           int squareHeight,
-                           int columnCount,
-                           int rowCount) {
-        this.squareWidth = squareWidth;
-        this.squareHeight = squareHeight;
+    @Override
+    public void setDimensions(int columnCount, int rowCount, int boxWidth, int boxHeight) {
+        this.boxWidth = boxWidth;
+        this.boxHeight = boxHeight;
         this.columnCount = columnCount;
         this.rowCount = rowCount;
+        super.setWidth(boxWidth * columnCount);
+        super.setHeight(boxHeight * rowCount);
     }
 
     private void paintLines(){
+        paintHorizontalLines();
+        paintVerticalLines();
+    }
+
+    private void paintVerticalLines() {
         for(int column = 0; column < columnCount; column++)
-            getGraphicsContext2D().strokeLine(0, column*squareHeight, getWidth(), column*squareWidth);
+            getGraphicsContext2D().strokeLine(column * boxWidth, 0,column * boxWidth, getHeight());
+    }
+
+    private void paintHorizontalLines() {
         for(int row = 0; row < rowCount; row++)
-            getGraphicsContext2D().strokeLine(row*squareHeight, 0,row*squareHeight, getHeight());
+            getGraphicsContext2D().strokeLine(0, row * boxHeight, getWidth(), row * boxHeight);
     }
 
-    private void paintSquare(int row, int column, Color color){
+    private void paintBox(int row, int column, Color color){
         getGraphicsContext2D().setFill(color);
-        getGraphicsContext2D().fillRect(row*squareHeight,column*squareWidth,squareHeight,squareWidth);
+        getGraphicsContext2D().fillRect(column * boxWidth,row * boxHeight, boxWidth, boxHeight);
     }
 
-
-
+    private void clearBox(int row, int column){
+        getGraphicsContext2D().clearRect(column * boxWidth,row * boxHeight, boxWidth, boxHeight);
+    }
 }
\ No newline at end of file

src/main/java/view/Grid.java

 package view;
 
-import javafx.scene.paint.Color;
 import javafx.util.Pair;
-import model.Position;
+import util.Position;
 
 import java.util.List;
 
+/**
+ * This interface represents a generic grid structure for displaying two-dimensional data.
+ *
+ * @param <E> The type of elements stored in the grid.
+ */
 public interface Grid<E> {
+
+  /**
+   * Repaint the grid with a list of elements, each associated with their respective positions.
+   *
+   * @param elements A list of pairs, each containing a position and the element to be displayed at that position.
+   */
   void repaint(List<Pair<Position, E>> elements);
+
+  /**
+   * Repaint the grid with a two-dimensional array of elements. The array's dimensions should match
+   * the row and column count of the grid.
+   *
+   * @param elements A two-dimensional array of elements to be displayed on the grid.
+   */
   void repaint(E[][] elements);
-  int getColumnCount();
-  int getRowCount();
+
+  /**
+   * Set the dimensions of the grid to the specified number of columns, number of rows, square width,
+   * and square height.
+   *
+   * @param columnCount The new number of columns in the grid.
+   * @param rowCount The new number of rows in the grid.
+   * @param squareWidth The width of each square within the grid.
+   * @param squareHeight The height of each square within the grid.
+   */
+  void setDimensions(int columnCount, int rowCount, int squareWidth, int squareHeight);
+
+  /**
+   * Get the number of columns in the grid.
+   *
+   * @return The number of columns in the grid.
+   */
+  int columnCount();
+
+  /**
+   * Get the number of rows in the grid.
+   *
+   * @return The number of rows in the grid.
+   */
+  int rowCount();
 }
+

src/main/resources/view/view.fxml

@@ -6,10 +6,20 @@
 <?import view.FirefighterGrid?>
 
 <?import javafx.scene.control.ToggleButton?>
+<?import javafx.scene.control.Separator?>
+<?import javafx.scene.control.Label?>
 <HBox styleClass="background" stylesheets="@DarkTheme.css"
       xmlns="http://javafx.com/javafx" xmlns:fx="http://javafx.com/fxml"
       fx:controller="controller.Controller">
   <VBox>
+    <Separator maxHeight="-Infinity" maxWidth="-Infinity"
+               prefHeight="24.0" prefWidth="200.0"/>
+    <Label maxHeight="-Infinity" maxWidth="-Infinity" alignment="CENTER" prefHeight="24.0" prefWidth="200.0"
+           text="Generation number"/>
+    <Label fx:id="generationNumberLabel" alignment="CENTER" contentDisplay="TEXT_ONLY"
+           maxHeight="-Infinity" maxWidth="-Infinity" prefHeight="24.0" prefWidth="200.0"/>
+    <Separator maxHeight="-Infinity" maxWidth="-Infinity"
+               prefHeight="24.0" prefWidth="200.0"/>
     <Button fx:id="restartButton" maxHeight="-Infinity" maxWidth="-Infinity"
             mnemonicParsing="false" onAction="#restartButtonAction" prefHeight="24.0" prefWidth="200.0"
             text="Restart"/>
@@ -23,7 +33,7 @@
                   mnemonicParsing="false" onAction="#pauseToggleButtonAction" prefHeight="24.0"
                   prefWidth="200.0" styleClass="button" text="Pause"/>
   </VBox>
-  <FirefighterGrid fx:id="grid" width="1000.0" height="1000.0"
+  <FirefighterGrid fx:id="grid"
                    xmlns="http://javafx.com/javafx"
                    xmlns:fx="http://javafx.com/fxml">
   </FirefighterGrid>

src/test/java/model/FirefighterBoardTest.java

+package model;
+
+import org.junit.jupiter.api.Test;
+import util.Position;
+
+import java.util.List;
+
+import static org.assertj.core.api.Assertions.*;
+
+public class FirefighterBoardTest {
+  @Test
+  void testColumnCount(){
+    Board<List<ModelElement>> board = new FirefighterBoard(20, 10, 1, 3);
+    assertThat(board.columnCount()).isEqualTo(20);
+  }
+  @Test
+  void testRowCount(){
+    Board<List<ModelElement>> board = new FirefighterBoard(20, 10, 1, 3);
+    assertThat(board.rowCount()).isEqualTo(10);
+  }
+  @Test
+  void testStepNumber(){
+    Board<List<ModelElement>> board = new FirefighterBoard(20, 10, 1, 3);
+    for(int index = 0; index < 10; index++){
+      assertThat(board.stepNumber()).isEqualTo(index);
+      board.updateToNextGeneration();
+    }
+    assertThat(board.stepNumber()).isEqualTo(10);
+  }
+  @Test
+  void testGetState_afterSet(){
+    Board<List<ModelElement>> board = new FirefighterBoard(20, 10, 0, 0);
+    Position position = new Position(1,2);
+    assertThat(board.getState(position)).isEmpty();
+    board.setState(List.of(ModelElement.FIRE), position);
+    assertThat(board.getState(position)).containsExactly(ModelElement.FIRE);
+  }
+
+}

src/test/java/view/FirefighterGridTest.java

+package view;
+
+import org.junit.jupiter.api.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class FirefighterGridTest {
+  @Test
+  void testColumnCount(){
+    Grid<ViewElement> grid = new FirefighterGrid();
+    grid.setDimensions(20,10,10,10);
+    assertThat(grid.columnCount()).isEqualTo(20);
+  }
+  @Test
+  void testRowCount(){
+    Grid<ViewElement> grid = new FirefighterGrid();
+    grid.setDimensions(20,10,10,10);
+    assertThat(grid.rowCount()).isEqualTo(10);
+  }
+}