public void TestMapInfectionCounter() { City newyork = map.addCity("ny", DiseaseColor.BLUE); City atl = map.addCity("atl", DiseaseColor.BLUE); City washington = map.addCity("washington", DiseaseColor.BLUE); City chicago = map.addCity("chicago", DiseaseColor.BLUE); City.makeAdjacent(newyork, atl); City.makeAdjacent(atl, washington); City.makeAdjacent(washington, chicago); map = map.addDisease(atl, 2); map = map.addDisease(chicago, 1); map = map.addDisease(newyork, 3); GameState gs = new GameState(newyork, map); Assert.AreEqual(6, gs.map.numInfectionsInCities); Map newMap = map.removeDisease(atl, DiseaseColor.BLUE); GameState newgs = new GameState(gs, newMap); Assert.AreEqual(5, newgs.map.numInfectionsInCities); }