new function to remove exposures

Author: gabrielaznar

climada/entity/exposures/base.py

@@ -173,6 +173,37 @@ def read(self, files, descriptions='', var_names=None):
         for file, desc, var in zip(all_files, desc_list, var_list):
             self.append(Exposures._read_one(file, desc, var))
 
+    def remove(self, exp_id):
+        """Remove one exposure with given id.
+
+        Parameters:
+            exp_id (list(int) or int): exposure ids.
+        """
+        if not isinstance(exp_id, list):
+            exp_id = [exp_id]
+        try:
+            pos_del = []
+            for one_id in exp_id:
+                pos_del.append(np.argwhere(self.id == one_id)[0][0])
+        except IndexError:
+            LOGGER.info('No exposure with id %s.', exp_id)
+            return
+        self.coord = np.delete(self.coord, pos_del, axis=0)
+        self.value = np.delete(self.value, pos_del)
+        self.impact_id = np.delete(self.impact_id, pos_del)
+        self.id = np.delete(self.id, pos_del)
+        if self.deductible.size:
+            self.deductible = np.delete(self.deductible, pos_del)
+        if self.cover.size:
+            self.cover = np.delete(self.cover, pos_del)
+        if self.category_id.size:
+            self.category_id = np.delete(self.category_id, pos_del)
+        if self.region_id.size:
+            self.region_id = np.delete(self.region_id, pos_del)
+        old_assigned = self.assigned.copy()
+        for key, val in old_assigned.items():
+            self.assigned[key] = np.delete(val, pos_del)
+
     def append(self, exposures):
         """Check and append variables of input Exposures to current Exposures.
 
@@ -276,6 +307,12 @@ def coord(self, value):
         else:
             self._coord = value
 
+    @property
+    def size(self):
+        """ Get longitude from coord array """
+        self.check()
+        return self.value.size
+
     @staticmethod
     def _read_one(file_name, description='', var_names=None):
         """Read one file and fill attributes.

climada/entity/exposures/test/test_base.py

@@ -220,6 +220,36 @@ def test_read_incompatible_fail(self):
         self.assertIn('Append not possible. Different reference years.', \
                          cm.output[0])
 
+class TestRemove(unittest.TestCase):
+    """Check read function with several files"""
+
+    def test_read_one_pass(self):
+        """Remove an exposure of existing id."""
+        expo = good_exposures()
+        expo.remove(1)
+        expo.check()
+        self.assertEqual(expo.size, 2)
+
+    def test_read_two_pass(self):
+        """Remove an exposure of existing id."""
+        expo = good_exposures()
+        expo.remove([1,3])
+        expo.check()
+        self.assertEqual(expo.size, 1)
+
+    def test_read_three_pass(self):
+        """Remove an exposure of existing id."""
+        expo = good_exposures()
+        expo.remove([1,3, 2])
+        expo.check()
+        self.assertEqual(expo.size, 0)
+
+    def test_read_wrong_one_pass(self):
+        """Remove exposure that does not exist."""
+        expo = good_exposures()
+        expo.remove(5)
+        self.assertEqual(expo.size, 3)
+
 class TestChecker(unittest.TestCase):
     """Test loading funcions from the Exposures class"""
     def test_check_wrongValue_fail(self):
@@ -323,4 +353,5 @@ def test_check_wrongId_fail(self):
 TESTS.addTests(unittest.TestLoader().loadTestsFromTestCase(TestAppend))
 TESTS.addTests(unittest.TestLoader().loadTestsFromTestCase(TestReadParallel))
 TESTS.addTests(unittest.TestLoader().loadTestsFromTestCase(TestConstructor))
+TESTS.addTests(unittest.TestLoader().loadTestsFromTestCase(TestRemove))
 unittest.TextTestRunner(verbosity=2).run(TESTS)