def test_layer_purpose_for_layer(self): """Test for purpose_for_layer method.""" expected = ['aggregation', 'exposure', 'hazard'] self.assertListEqual(expected, purposes_for_layer('polygon')) expected = ['exposure', 'hazard'] self.assertListEqual(expected, purposes_for_layer('raster')) expected = ['exposure'] self.assertListEqual(expected, purposes_for_layer('line'))
def purposes_for_layer(self): """Return a list of valid purposes for the current layer. :returns: A list where each value represents a valid purpose. :rtype: list """ layer_geometry_key = self.parent.get_layer_geometry_key() return purposes_for_layer(layer_geometry_key)