def test_main(self):
named_ops = [(str(idx), self.Operator()) for idx in range(3)]
op_container = ops.OperatorContainer(named_ops)
actuals = op_container.named_children()
desireds = named_ops
assert_named_modules_identical(actuals, desireds)
def test_MultiOperatorLoss():
class TestOperator(ops.Operator):
def process_input_image(self, image):
pass
named_ops = [(str(idx), TestOperator()) for idx in range(3)]
multi_op_loss = loss.MultiOperatorLoss(named_ops)
actuals = multi_op_loss.named_children()
desireds = named_ops
assert_named_modules_identical(actuals, desireds)
def test_OperatorContainer():
class TestOperator(ops.Operator):
def process_input_image(self, image):
pass
named_ops = [(str(idx), TestOperator()) for idx in range(3)]
op_container = ops.OperatorContainer(named_ops)
actuals = op_container.named_children()
desireds = named_ops
assert_named_modules_identical(actuals, desireds)
def test_MultiLayerEncoder_named_children_from():
layers = [str(idx) for idx in range(3)]
modules = [(layer, nn.Module()) for layer in layers]
multi_layer_encoder = enc.MultiLayerEncoder(modules)
actuals = multi_layer_encoder.named_children_from(layers[-2])
desireds = modules[1:]
assert_named_modules_identical(actuals, desireds)
actuals = multi_layer_encoder.named_children_from(layers[-2], include_first=False)
desireds = modules[2:]
assert_named_modules_identical(actuals, desireds)