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)