implement BoxCell::remove()

STEllAR-GROUP · Aug 15, 2016 · 165848e · 165848e
1 parent af82749
commit 165848e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/storage/boxcell.h b/src/storage/boxcell.h
@@ -81,6 +81,11 @@ class BoxCell
         particles << particle;
     }
 
+    inline void remove(const std::size_t i)
+    {
+        particles.remove(i);
+    }
+
     inline std::size_t size() const
     {
         return particles.size();

diff --git a/src/storage/test/unit/boxcelltest.h b/src/storage/test/unit/boxcelltest.h
@@ -394,6 +394,21 @@ class BoxCellTest : public CxxTest::TestSuite
         }
     }
 
+    void testRemove()
+    {
+        typedef BoxCell<FixedArray<SpawningParticle, 222> > CellType;
+
+        FloatCoord<3> origin(3.0, 3.0, 3.0);
+        FloatCoord<3> cellDim(2.0, 2.0, 2.0);
+
+        CellType cell(origin, cellDim);
+        cell.insert(SpawningParticle(FloatCoord<3>(3.5, 3.5, 3.5), 100));
+
+        TS_ASSERT_EQUALS(cell.size(), 1);
+        cell.remove(0);
+        TS_ASSERT_EQUALS(cell.size(), 0);
+    }
+
 private:
     Coord<2> gridDim;
     FloatCoord<2> cellDim;