def test_read_area_source_model(self): area_source = create_area_source() as_reader = NRMLReader(AREA_SOURCE, SCHEMA) area_source_gen = as_reader.read() read_first_area_source = area_source_gen.next() self.assertEqual(area_source, read_first_area_source)
def read_source_model(context): """ Create source model definitions by reading a source model. :param context: shared datastore across different jobs in a pipeline """ sm_definitions = [] reader = NRMLReader(context.config["source_model_file"], NRML_SCHEMA_PATH) for sm in reader.read(): sm_definitions.append(sm) context.sm_definitions = sm_definitions LOGGER.debug("* Eq number source models: %s" % len(context.sm_definitions))
def read_source_model(context): """ Create source model definitions by reading a source model. :param context: shared datastore across different jobs in a pipeline """ sm_definitions = [] reader = NRMLReader(context.config['source_model_file'], NRML_SCHEMA_PATH) for sm in reader.read(): sm_definitions.append(sm) context.sm_definitions = sm_definitions LOGGER.debug("* Eq number source models: %s" % len(context.sm_definitions))
def test_generated_area_sources(self): num_expected_area_sources = 2 as_reader = NRMLReader(AREA_SOURCES, SCHEMA) for num_area_sources, _ in enumerate(as_reader.read(), start=1): pass self.assertEqual(num_expected_area_sources, num_area_sources)