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 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)
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)
