如何让Pylance识别抽象基类最终继承自unittest.TestCase？

问题背景 #

你将两个接口实现的公共单元测试逻辑放在抽象基类CommonScenarios中，让具体测试类同时继承该基类与unittest.TestCase。但Pylance无法识别基类中self.assertEqual等unittest断言方法，因为CommonScenarios本身没有直接继承unittest.TestCase。

解决方案 #

这里有几种实用的解决方式：

1. 让抽象基类继承unittest.TestCase #

直接修改CommonScenarios的继承关系，让它继承unittest.TestCase，这样Pylance能直接识别断言方法。具体修改如下：

import abc
import unittest
import example

class CommonScenarios(abc.ABC, unittest.TestCase):
    def test_do_something(self):
        expected = self._implementation.doSomething()
        actual = 987
        self.assertEqual(expected, actual)

class TestImplementationA(CommonScenarios):
    def setUp(self):
        self._implementation = example.ImplementationA(True, 123)

class TestImplementationB(CommonScenarios):
    def setUp(self):
        self._implementation = example.ImplementationB()

这样具体测试类只需继承CommonScenarios即可，既保留了公共逻辑，也解决了Pylance的警告。

2. 用类型提示标注self的类型 #

如果不想让基类继承unittest.TestCase，可以通过类型提示告诉Pylance，self的类型会是unittest.TestCase的子类。

方式A（Python 3.11+）

import abc
import unittest
from typing import Self
import example

class CommonScenarios(abc.ABC):
    def test_do_something(self: Self) -> None:
        expected = self._implementation.doSomething()
        actual = 987
        self.assertEqual(expected, actual)

class TestImplementationA(CommonScenarios, unittest.TestCase):
    def setUp(self):
        self._implementation = example.ImplementationA(True, 123)

class TestImplementationB(CommonScenarios, unittest.TestCase):
    def setUp(self):
        self._implementation = example.ImplementationB()

方式B（兼容旧版本Python）

import abc
import unittest
from typing import TypeVar
import example

T = TypeVar('T', bound='CommonScenarios')

class CommonScenarios(abc.ABC):
    def test_do_something(self: T) -> None:
        expected = self._implementation.doSomething()
        actual = 987
        self.assertEqual(expected, actual)

class TestImplementationA(CommonScenarios, unittest.TestCase):
    def setUp(self):
        self._implementation = example.ImplementationA(True, 123)

class TestImplementationB(CommonScenarios, unittest.TestCase):
    def setUp(self):
        self._implementation = example.ImplementationB()

3. 忽略Pylance警告（不推荐） #

如果只是想快速消除警告，可以在断言方法调用前添加注释：

# pyright: ignore[reportAttributeAccessIssue]
self.assertEqual(expected, actual)

但这种方法只是屏蔽警告，没有从根本上解决类型识别问题，不建议长期使用。

总结 #

最简洁可靠的方式是让抽象基类继承unittest.TestCase，这样既符合测试类的继承逻辑，又能让Pylance正确识别所有unittest方法。如果有特殊原因不能让基类继承TestCase，类型提示的方式是更好的选择。

内容的提问来源于stack exchange，提问作者Arseni Mourzenko