本文整理汇总了Python中gnome.spill_container.SpillContainer.itersubstancedata方法的典型用法代码示例。如果您正苦于以下问题：Python SpillContainer.itersubstancedata方法的具体用法？Python SpillContainer.itersubstancedata怎么用？Python SpillContainer.itersubstancedata使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类gnome.spill_container.SpillContainer的用法示例。
在下文中一共展示了SpillContainer.itersubstancedata方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_no_substance
# 需要导入模块: from gnome.spill_container import SpillContainer [as 别名]
# 或者: from gnome.spill_container.SpillContainer import itersubstancedata [as 别名]
def test_no_substance(self):
'''
no substance means run trajectory without an OilProps object/without
weathering is one reason to do this
'''
sc = SpillContainer()
sc.spills += [Spill(Release(datetime.now(), 10),
element_type=floating(substance=None),
name='spill0'),
Spill(Release(datetime.now(), 10),
element_type=floating(substance=None),
name='spill1')]
assert len(sc.itersubstancedata('mass')) == 0
assert len(sc.get_substances()) == 1
assert len(sc.get_substances(complete=False)) == 0
# iterspillsbysubstance() iterates through all the spills associated
# with each substance including the spills where substance is None
assert len(sc.iterspillsbysubstance()) == 2示例2: test_spills_different_substance_release
# 需要导入模块: from gnome.spill_container import SpillContainer [as 别名]
# 或者: from gnome.spill_container.SpillContainer import itersubstancedata [as 别名]
def test_spills_different_substance_release(self):
'''
Test data structure gets correctly set/updated after release_elements
is invoked
'''
sc = SpillContainer()
rel_time = datetime(2014, 1, 1, 12, 0, 0)
end_time = rel_time + timedelta(hours=1)
time_step = 900
splls0 = [point_line_release_spill(100, (1, 1, 1),
rel_time,
end_release_time=end_time,
element_type=floating(substance=test_oil),
amount=100,
units='kg'),
point_line_release_spill(50, (2, 2, 2),
rel_time + timedelta(seconds=900),
element_type=floating(substance=test_oil),
amount=150,
units='kg'),
]
sc.spills += splls0
splls1 = point_line_release_spill(10, (0, 0, 0),
rel_time,
element_type=floating(substance=None))
sc.spills += splls1
at = {'density', 'mass_components'}
sc.prepare_for_model_run(at)
assert len(sc.get_substances()) == 2
print '\nElements released:'
for ix in range(-1, 8):
time = rel_time + timedelta(seconds=time_step) * ix
num_rel = sc.release_elements(time_step, time)
print num_rel
for substance, data in sc.itersubstancedata(at):
assert substance.name == test_oil
idx = sc._substances_spills.substances.index(substance)
mask = sc['substance'] == idx
for array in at:
assert array in data
assert np.all(data[array] == sc[array][mask])