def test_add_operator_type_not_in_registry(self, operator_class, return_value): assert add_operator_type(operator_class) is return_value
def test_add_operator_type(self): class Mock(): display_name = 'Mock operator' assert add_operator_type(Mock) is True