def METHOD_NAME(self):
    model = self.model_context.model
    wrap_cls = [ToyCustomModule]
    result = to_module_class_by_name(model, wrap_cls)
    self.assertEqual(len(result), 1)
    wrap_cls = ["ToyCustomModule"]
    result = to_module_class_by_name(model, wrap_cls)
    self.assertEqual(len(result), 1)
    self.assertEqual(result[0], ToyCustomModule)
    wrap_cls = (ToyCustomModule, "Linear", "bad_module")
    result = to_module_class_by_name(model, wrap_cls)
    self.assertEqual(len(result), 2)
    self.assertTrue(isinstance(result, tuple))
    self.assertEqual(result[0], ToyCustomModule)
    self.assertEqual(result[1].__name__, "Linear")