Grid: Implement a set operator for array access

Author: rhaendel

src/main/kotlin/de/ronny_h/aoc/extensions/grids/Grid.kt

@@ -76,9 +76,13 @@ abstract class Grid<T>(
             ?: fallbackElement
     }
 
+    operator fun set(row: Int, col: Int, value: T) {
+        grid[row][col] = value
+    }
+
     fun getAt(position: Coordinates) = get(position.row, position.col)
     fun setAt(position: Coordinates, element: T) {
-        grid[position.row][position.col] = element
+        this[position.row, position.col] = element
     }
 
     /**

src/test/kotlin/de/ronny_h/aoc/extensions/grids/GridTest.kt

@@ -88,6 +88,15 @@ class GridTest : StringSpec() {
             grid[1, 1] shouldBe '4'
         }
 
+        "set with array access sets the element at the given coordinates" {
+            val grid = SimpleCharGrid(listOf("12", "34"))
+            grid[1, 0] = '5'
+            grid[0, 0] shouldBe '1'
+            grid[0, 1] shouldBe '2'
+            grid[1, 0] shouldBe '5'
+            grid[1, 1] shouldBe '4'
+        }
+
         "subGridAt returns a sub grid at the given coordinates" {
             val input = """
                 1234