Ejemplo n.º 1
0
    def test_my_inputs(self):
        storage = StorageFactory().get_storage('dict_storage')

        landuse_filter_table_name = 'landuse_filter'
        storage.write_table(
                table_name=landuse_filter_table_name,
                table_data={
                    'id':array([1,2,3,4]),
                    'from_development_type_id':array([1,1,3,5]),
                    'to_development_type_id':array([4,6,2,2])
                    },
            )

        landuse_filter = Dataset(
            id_name='id',
            in_storage=storage,
            in_table_name = landuse_filter_table_name
            )

        values = VariableTestToolbox().compute_variable(self.variable_name,
            data_dictionary = {
                'gridcell':{
                    'development_type_id':array([1,2,5])
                    },
                'development_filter':landuse_filter
                },
            dataset = 'gridcell'
            )

        should_be = array([1, 0, 1])

        self.assert_(ma.allequal(values, should_be),
            'Error in ' + self.variable_name)