README.md

@@ -116,10 +116,10 @@ Disse siste delene med informasjon vil ikke endre seg særlig fra kall til kall,
 I klassen `CellPositionToPixelConverter`:
 
 * [ ] Opprett instansvariabler:
-  * Et `Rectangle2D` -objekt `box` som beskriver innenfor hvilket område rutenettet skal tegnes
-  * Et `GridDimension` -objekt `gd` som beskriver størrelsen til rutenettet rutene vil være en del av
+  * En `Rectangle2D` -variabel `box` som beskriver innenfor hvilket område rutenettet skal tegnes
+  * En `GridDimension` -variabel `gd` som beskriver størrelsen til rutenettet rutene vil være en del av
   * En `double` kalt `margin` som beskriver hvor stor avstanden skal være mellom rutene
-* [ ] Opprett en konstruktør i klassen med tre parametre: et `Rectangle2D` -objekt, et `GridDimension` -objekt og en `double`. Initaliser feltvariablene med argumentene som mottas i konstruktøren.
+* [ ] Opprett en konstruktør i klassen med tre parametre: en `Rectangle2D` -variabel, en `GridDimension` -variabel og en `double`. Initaliser feltvariablene med argumentene som mottas i konstruktøren.
 * [ ] Opprett metoden `getBoundsForCell` med en parameter av typen `CellPosition` (i figur under navgitt `cp`) og returtype `Rectangle2D`.
 
 Returverdien er et `Rectangle2D` -objekt. For å opprette dette objektet, må du regne ut fire verdier: x, y, bredde og høyde for den gitte ruten. Så kan du returnere et nytt `Rectangle2D.Double` -objekt med disse verdiene.
@@ -136,7 +136,7 @@ Hint:
 > Eksempel: anta at du får følgende parametre (som i illustrasjonen):
 >  * `box.getX()` er 30 og `box.getY()` er 30
 >  * `box.getWidth()` er 340 og `box.getHeight()` er 240
->  * `cellPosition.col()` er 2 og `cellPosition.row()` er 1
+>  * `cp.col()` er 2 og `cp.row()` er 1
 >  * `gd.cols()` er 4 og `gd.rows()` er 3
 >  * `margin` er 30
 >

src/test/java/no/uib/inf101/colorgrid/TestColorGrid.java

@@ -25,6 +25,10 @@ public class TestColorGrid {
     grid.set(new CellPosition(1, 2), Color.RED);
     assertEquals(Color.RED, grid.get(new CellPosition(1, 2)));
     assertEquals(null, grid.get(new CellPosition(2, 1)));
+
+    // Check that we can set a new value and retrieve it again
+    grid.set(new CellPosition(1, 2), Color.GREEN);
+    assertEquals(Color.GREEN, grid.get(new CellPosition(1, 2)));
   }
 
   @Test
@@ -87,6 +91,28 @@ public class TestColorGrid {
     }
   }
 
+  @Test
+  public void testGetCellsWithNull() {
+    IColorGrid grid = new WColorGrid(3, 1);
+
+    grid.set(new CellPosition(2, 0), Color.GREEN);
+
+    List<CellColor> expected = Arrays.asList(
+        new CellColor(new CellPosition(0, 0), null),
+        new CellColor(new CellPosition(1, 0), null),
+        new CellColor(new CellPosition(2, 0), Color.GREEN)
+    );
+    List<CellColor> actual = grid.getCells();
+
+    assertEquals(expected.size(), actual.size());
+    for (CellColor cp : actual) {
+      assertTrue(expected.contains(cp));
+    }
+    for (CellColor cp : expected) {
+      assertTrue(actual.contains(cp));
+    }
+  }
+
   @Test
   public void testIndexOutOfBoundsException() {
     IColorGrid grid = new WColorGrid(10, 20);