diff --git a/tdt4140-gr1800/app.core/src/test/java/tdt4140/gr1800/app/core/GeoLocationsTest.java b/tdt4140-gr1800/app.core/src/test/java/tdt4140/gr1800/app/core/GeoLocationsTest.java index 271248bb5d264f096a47977f3c2361b25e3aaf4a..1cdd829250a43a5d2473dbaeaae029082cf334b5 100644 --- a/tdt4140-gr1800/app.core/src/test/java/tdt4140/gr1800/app/core/GeoLocationsTest.java +++ b/tdt4140-gr1800/app.core/src/test/java/tdt4140/gr1800/app/core/GeoLocationsTest.java @@ -70,5 +70,19 @@ public class GeoLocationsTest { geoLocations.addLocation(latLong); LatLong nearestlocations = geoLocations.findNearestLocation(latLong); Assert.assertEquals(latLong, nearestlocations); - } + } + + @Test + public void testRemoveSameLocations() { + GeoLocations geoLocations1 = new GeoLocations(new LatLong(0, 0), new LatLong(1, 1)); + Assert.assertEquals(2, geoLocations1.size()); + geoLocations1.removeLocations(new LatLong(0, 0), 0.0); + Assert.assertEquals(1, geoLocations1.size()); + geoLocations1.removeLocations(new LatLong(1, 1), 0.0); + Assert.assertEquals(0, geoLocations1.size()); + GeoLocations geoLocations2 = new GeoLocations(new LatLong(0, 0), new LatLong(0, 0)); + Assert.assertEquals(2, geoLocations2.size()); + geoLocations2.removeLocations(new LatLong(0, 0), 0.0); + Assert.assertEquals(0, geoLocations2.size()); + } }