added test for another edge condition

bagherilab · Dec 19, 2024 · b18a4ac · b18a4ac
1 parent 2f107f3
commit b18a4ac
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 3 deletions.
diff --git a/test/arcade/patch/agent/cell/PatchCellTest.java b/test/arcade/patch/agent/cell/PatchCellTest.java
@@ -263,8 +263,7 @@ public void checkLocation_locationEmpty_returnTrue() {
         assertEquals(true, actual);
     }
 
-    final Bag createPatchCellsWithVolumeAndCriticalHeight(
-            int n, double volume, double critHeight) {
+    final Bag createPatchCellsWithVolumeAndCriticalHeight(int n, double volume, double critHeight) {
         Bag bag = new Bag();
         for (int i = 0; i < n; i++) {
             PatchCellContainer container =

diff --git a/test/arcade/patch/agent/cell/PatchCellTissueTest.java b/test/arcade/patch/agent/cell/PatchCellTissueTest.java
@@ -87,7 +87,7 @@ final Bag createPatchTissueCellsWithVolumeAndCriticalHeight(
     }
 
     @Test
-    public void findFreeLocations_roomForMultipleCancerCells_returnsCurrentAndOpenLocation() {
+    public void findFreeLocations_exceedsMaxDensitys_returnsOnlyCurrentLocation() {
         doReturn(0.0).when(parametersMock).getDouble(anyString());
         doReturn(0).when(parametersMock).getInt(anyString());
 
@@ -123,4 +123,42 @@ public void findFreeLocations_roomForMultipleCancerCells_returnsCurrentAndOpenLo
         assertTrue(freeLocations.contains(locationMock));
         assertFalse(freeLocations.contains(notFreeLocation));
     }
+
+    @Test
+    public void findFreeLocations_containsAnotherPopulation_returnsFreeAndCurrentLocation() {
+        doReturn(0.0).when(parametersMock).getDouble(anyString());
+        doReturn(0).when(parametersMock).getInt(anyString());
+
+        doReturn(1000.).when(locationMock).getVolume();
+        doReturn(100.).when(locationMock).getArea();
+
+        PatchLocation notFreeLocation = mock(PatchLocation.class);
+        Bag notFreeBag = createPatchTissueCellsWithVolumeAndCriticalHeight(1, 200, 10);
+        doReturn(notFreeBag).when(gridMock).getObjectsAtLocation(notFreeLocation);
+        doReturn(1000.).when(notFreeLocation).getVolume();
+        doReturn(100.).when(notFreeLocation).getArea();
+        ArrayList<Location> neighborLocations = new ArrayList<>();
+        neighborLocations.add(notFreeLocation);
+        doReturn(neighborLocations).when(locationMock).getNeighbors();
+
+        PatchCellContainer container =
+                new PatchCellContainer(
+                        cellID,
+                        cellParent,
+                        cellPop - 1,
+                        cellAge,
+                        cellDivisions,
+                        cellState,
+                        cellVolume,
+                        cellHeight,
+                        cellCriticalVolume,
+                        cellCriticalHeight);
+        PatchCell cell = new PatchCellTissue(container, locationMock, parametersMock);
+
+        Bag freeLocations = cell.findFreeLocations(simMock, false);
+
+        assertEquals(2, freeLocations.size());
+        assertTrue(freeLocations.contains(locationMock));
+        assertTrue(freeLocations.contains(notFreeLocation));
+    }
 }