Python - Testing an abstract base class

前端 未结 6 1323
醉话见心
醉话见心 2020-12-04 16:27

I am looking for ways / best practices on testing methods defined in an abstract base class. One thing I can think of directly is performing the test on all concrete subclas

相关标签:
6条回答
  • 2020-12-04 17:05

    As properly put by lunaryon, it is not possible. The very purpose of ABCs containing abstract methods is that they are not instantiatable as declared.

    However, it is possible to create a utility function that introspects an ABC, and creates a dummy, non abstract class on the fly. This function could be called directly inside your test method/function and spare you of having to wite boiler plate code on the test file just for testing a few methods.

    def concreter(abclass):
        """
        >>> import abc
        >>> class Abstract(metaclass=abc.ABCMeta):
        ...     @abc.abstractmethod
        ...     def bar(self):
        ...        return None
    
        >>> c = concreter(Abstract)
        >>> c.__name__
        'dummy_concrete_Abstract'
        >>> c().bar() # doctest: +ELLIPSIS
        (<abc_utils.Abstract object at 0x...>, (), {})
        """
        if not "__abstractmethods__" in abclass.__dict__:
            return abclass
        new_dict = abclass.__dict__.copy()
        for abstractmethod in abclass.__abstractmethods__:
            #replace each abc method or property with an identity function:
            new_dict[abstractmethod] = lambda x, *args, **kw: (x, args, kw)
        #creates a new class, with the overriden ABCs:
        return type("dummy_concrete_%s" % abclass.__name__, (abclass,), new_dict)
    
    0 讨论(0)
  • 2020-12-04 17:06

    In newer versions of Python you can use unittest.mock.patch()

    class MyAbcClassTest(unittest.TestCase):
    
        @patch.multiple(MyAbcClass, __abstractmethods__=set())
        def test(self):
             self.instance = MyAbcClass() # Ha!
    
    0 讨论(0)
  • 2020-12-04 17:06

    You can use multiple inheritance practice to have access to the implemented methods of the abstract class. Obviously following such design decision depends on the structure of the abstract class since you need to implement abstract methods (at least bring the signature) in your test case.

    Here is the example for your case:

    class Abstract(object):
    
        __metaclass__ = abc.ABCMeta
    
        @abc.abstractproperty
        def id(self):
            return
    
        @abc.abstractmethod
        def foo(self):
            print("foo")
    
        def bar(self):
            print("bar")
    
    class AbstractTest(unittest.TestCase, Abstract):
    
        def foo(self):
            pass
        def test_bar(self):
            self.bar()
            self.assertTrue(1==1)
    
    0 讨论(0)
  • 2020-12-04 17:24

    Here is what I have found: If you set __abstractmethods__ attribute to be an empty set you'll be able to instantiate abstract class. This behaviour is specified in PEP 3119:

    If the resulting __abstractmethods__ set is non-empty, the class is considered abstract, and attempts to instantiate it will raise TypeError.

    So you just need to clear this attribute for the duration of tests.

    >>> import abc
    >>> class A(metaclass = abc.ABCMeta):
    ...     @abc.abstractmethod
    ...     def foo(self): pass
    

    You cant instantiate A:

    >>> A()
    Traceback (most recent call last):
    TypeError: Can't instantiate abstract class A with abstract methods foo
    

    If you override __abstractmethods__ you can:

    >>> A.__abstractmethods__=set()
    >>> A() #doctest: +ELLIPSIS
    <....A object at 0x...>
    

    It works both ways:

    >>> class B(object): pass
    >>> B() #doctest: +ELLIPSIS
    <....B object at 0x...>
    
    >>> B.__abstractmethods__={"foo"}
    >>> B()
    Traceback (most recent call last):
    TypeError: Can't instantiate abstract class B with abstract methods foo
    

    You can also use unittest.mock (from 3.3) to override temporarily ABC behaviour.

    >>> class A(metaclass = abc.ABCMeta):
    ...     @abc.abstractmethod
    ...     def foo(self): pass
    >>> from unittest.mock import patch
    >>> p = patch.multiple(A, __abstractmethods__=set())
    >>> p.start()
    {}
    >>> A() #doctest: +ELLIPSIS
    <....A object at 0x...>
    >>> p.stop()
    >>> A()
    Traceback (most recent call last):
    TypeError: Can't instantiate abstract class A with abstract methods foo
    
    0 讨论(0)
  • 2020-12-04 17:24

    Perhaps a more compact version of the concreter proposed by @jsbueno could be:

    def concreter(abclass):
        class concreteCls(abclass):
            pass
        concreteCls.__abstractmethods__ = frozenset()
        return type('DummyConcrete' + abclass.__name__, (concreteCls,), {})
    

    The resulting class still has all original abstract methods (which can be now called, even if this is not likely to be useful...) and can be mocked as needed.

    0 讨论(0)
  • 2020-12-04 17:26

    No, it's not. The very purpose of abc is to create classes that cannot be instantiated unless all abstract attributes are overridden with concrete implementations. Hence you need to derive from the abstract base class and override all abstract methods and properties.

    0 讨论(0)
提交回复
热议问题