def pickMultiple_graph():
graph = Graph(name='graph')
graph.add(
PickN(name='cspad_pickN',
N=9,
inputs=['cspad', 'delta_t'],
outputs=['ncspads']))
graph.compile(num_workers=4, num_local_collectors=2)
return graph
def psana_graph(tmpdir_factory):
graph = Graph(name='graph')
graph.add(
PickN(name='picker', inputs=['xppcspad:raw:image'],
outputs=['picked']))
fname = tmpdir_factory.mktemp("psana_graph",
False).join("psana_graph.dill")
with open(fname, 'wb') as fd:
dill.dump(graph, fd)
return fname
def test_add(complex_graph):
complex_graph.compile(num_workers=4, num_local_collectors=2)
complex_graph.add(PickN(name='pickReferenceOne',
inputs=['BinningOff.Bins', 'BinningOff.Counts'],
outputs=['referenceOne']))
complex_graph.compile(num_workers=4, num_local_collectors=2)
complex_graph({'cspad': np.ones((200, 200)), 'laser': False, 'delta_t': 4}, color='worker')
worker = complex_graph({'cspad': np.ones((200, 200)), 'laser': False, 'delta_t': 5}, color='worker')
complex_graph(worker, color='localCollector')
localCollector = complex_graph(worker, color='localCollector')
globalCollector = complex_graph(localCollector, color='globalCollector')
assert worker == {'BinningOff_reduce_count_worker': {4: (10000.0, 1), 5: (10000.0, 1)}}
assert localCollector == {'BinningOff_reduce_count_localCollector': {4: (20000.0, 2), 5: (20000.0, 2)}}
np.testing.assert_equal(globalCollector['referenceOne'][0], np.array([4, 5]))
np.testing.assert_equal(globalCollector['referenceOne'][1], np.array([10000., 10000.]))
def eb_graph():
graph = Graph(name='graph')
graph.add(
PickN(name='pick1_values', N=1, inputs=['values'], outputs=['value']))
return dill.dumps(graph)