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