def test_get_all_sinks_and_sources(layers):
forced_orders = get_forced_orders(layers)
connections = get_connections(layers)
layout = create_layout_stub(layers)
all_sources = get_all_sources(forced_orders, connections, layout)
assert all_sources == [
'Input.outputs.default',
'Input.output_deltas.default',
'A.outputs.default',
('A.parameters.W', 'A.parameters.bias'),
'A.output_deltas.default',
('A.gradients.W', 'A.gradients.bias'),
'B.outputs.default',
('B.parameters.W', 'B.parameters.bias'),
'B.output_deltas.default',
('B.gradients.W', 'B.gradients.bias'),
'C.outputs.default',
('C.parameters.W', 'C.parameters.bias'),
'C.output_deltas.default',
('C.gradients.W', 'C.gradients.bias'),
'D.outputs.default',
('D.parameters.W', 'D.parameters.bias'),
'D.output_deltas.default',
('D.gradients.W', 'D.gradients.bias')]
def test_get_all_sinks_and_sources(layers):
forced_orders = get_forced_orders(layers)
connections = get_connections(layers)
layout = create_layout_stub(layers)
all_sources = get_all_sources(forced_orders, connections, layout)
assert all_sources == [
'Input.outputs.default',
'Input.output_deltas.default',
'A.outputs.default',
('A.parameters.W', 'A.parameters.bias'),
'A.output_deltas.default',
('A.gradients.W', 'A.gradients.bias'),
'B.outputs.default',
('B.parameters.W', 'B.parameters.bias'),
'B.output_deltas.default',
('B.gradients.W', 'B.gradients.bias'),
'C.outputs.default',
('C.parameters.W', 'C.parameters.bias'),
'C.output_deltas.default',
('C.gradients.W', 'C.gradients.bias'),
'D.outputs.default',
('D.parameters.W', 'D.parameters.bias'),
'D.output_deltas.default',
('D.gradients.W', 'D.gradients.bias')]
def test_get_forced_orders(layers):
assert get_forced_orders(layers) == [
('A.parameters.W', 'A.parameters.bias'),
('B.parameters.W', 'B.parameters.bias'),
('C.parameters.W', 'C.parameters.bias'),
('D.parameters.W', 'D.parameters.bias'),
('A.gradients.W', 'A.gradients.bias'),
('B.gradients.W', 'B.gradients.bias'),
('C.gradients.W', 'C.gradients.bias'),
('D.gradients.W', 'D.gradients.bias')
]
def test_get_forced_orders(layers):
assert get_forced_orders(layers) == [
('A.parameters.W', 'A.parameters.bias'),
('B.parameters.W', 'B.parameters.bias'),
('C.parameters.W', 'C.parameters.bias'),
('D.parameters.W', 'D.parameters.bias'),
('A.gradients.W', 'A.gradients.bias'),
('B.gradients.W', 'B.gradients.bias'),
('C.gradients.W', 'C.gradients.bias'),
('D.gradients.W', 'D.gradients.bias')
]
