| from pychecker2.TestSupport import WarningTester |
| from pychecker2 import ClassChecks |
| |
| class ClassTestCase(WarningTester): |
| def testClassChecks(self): |
| self.silent('class A:\n' |
| ' def f(self):\n' |
| ' return 0\n' |
| 'class B(A):\n' |
| ' def g(self):\n' |
| ' return self.f\n') |
| def testSignature(self): |
| w = ClassChecks.AttributeCheck.signatureChanged |
| self.silent('class Base:\n' |
| ' def f(self): pass\n' |
| 'class Derived(Base):\n' |
| ' def f(self): pass\n') |
| self.warning('class Base:\n' |
| ' def f(self): pass\n' |
| 'class Derived(Base):\n' |
| ' def f(self, x): pass\n', |
| 4, w, 'f', 'Base') |
| self.silent('class Base:\n' |
| ' def f(self, x, y, *args): pass\n' |
| 'class Derived(Base):\n' |
| ' def f(self, x, *args): pass\n') |
| self.warning('class Base:\n' |
| ' def f(self, x, *args): pass\n' |
| 'class Derived(Base):\n' |
| ' def f(self, x, y, *args): pass\n', |
| 4, w, 'f', 'Base') |
| self.silent('class Base:\n' |
| ' def __private(self, x, y): pass\n' |
| 'class Derived(Base):\n' |
| ' def __private(self, x): pass\n') |
| self.warning('class Base:\n' |
| ' def __private(self, x, y): pass\n' |
| 'class Derived(Base):\n' |
| ' def _Base__private(self, x): pass\n', |
| 4, w, '__private', 'Base') |
| |
| def testUnknownAttribute(self): |
| w = ClassChecks.AttributeCheck.unknownAttribute |
| self.warning('class C:\n' |
| ' def f(self): return self.x\n', 2, w, 'C', 'x') |
| self.silent('class C:\n' |
| ' def f(self): return self.x\n' |
| ' def __init__(self, v): self.x = v\n') |
| self.silent('class D:\n' |
| ' def __init__(self): self.x = 1\n' |
| 'class C(D):\n' |
| ' def f(self): return self.x\n') |
| self.silent('from pychecker2.utest.data import *\n' |
| 'class C(Data):\n' |
| ' def get_value2(self):\n' |
| ' return self.get_value()\n' |
| '\n') |
| imp = '\n\nimport' |
| frm = '\n\nfrom' |
| import_abuse = [ |
| (imp, 'pychecker2.utest.data', 'pychecker2.utest.data.Data'), |
| (imp, 'pychecker2.utest.data as F', 'F.Data'), |
| (frm, 'pychecker2.utest import data', 'data.Data'), |
| (frm, 'pychecker2.utest import data as F', 'F.Data'), |
| (frm, 'pychecker2.utest.data import Data', 'Data'), |
| (frm, 'pychecker2.utest.data import Data as F', 'F'), |
| (frm, 'pychecker2.utest.data import *', 'Data'), |
| ('class A:\n class B:\n import pychecker2.utest.data', |
| '', |
| 'A.B.pychecker2.utest.data.Data'), |
| ('class A:\n' |
| ' class B:\n' |
| ' from pychecker2.utest import data as F', |
| '', |
| 'A.B.F.Data') |
| ] |
| for fmt in import_abuse: |
| self.silent('%s %s\n' |
| 'class C(%s):\n' |
| ' def f(self): return self.value\n' % fmt) |
| self.silent('%s %s\n' |
| 'def g(v):\n' |
| ' class C(%s):\n' |
| ' def f(self): return self.value\n' |
| ' return C(v)\n' % fmt) |
| self.warning('%s %s\n' |
| 'class C(%s):\n' |
| ' def f(self): self.get_value = 1\n' % fmt, |
| 5, ClassChecks.AttributeCheck.methodRedefined, |
| 'get_value', 'C') |
| |
| self.warning('%s %s\n' |
| 'def g(v):\n' |
| ' class C(%s):\n' |
| ' def f(self): self.get_value = 1\n' |
| ' return C(v)\n' % fmt, |
| 6, ClassChecks.AttributeCheck.methodRedefined, |
| 'get_value', 'C') |
| |
| self.warning('%s %s\n' |
| 'class C(%s):\n' |
| ' def get_value(self, x): pass\n' % fmt, |
| 5, ClassChecks.AttributeCheck.signatureChanged, |
| 'get_value', 'Data') |
| |
| def testSpecial(self): |
| for fmt in [('__del__', 'self, args', 1), |
| ('__cmp__', 'self', 2), |
| ]: |
| self.warning('class C:\n' |
| ' def %s(%s):\n' |
| ' pass\n' % fmt[0:2], |
| 2, ClassChecks.SpecialCheck.specialMethod, |
| fmt[0], fmt[2], fmt[2] > 1 and "s" or "") |
| for fmt in [('__del__', 'self'), |
| ('__ge__', 'self, a, b = None, c = None'), |
| ]: |
| self.silent('class C:\n' |
| ' def %s(%s):\n' |
| ' pass\n' % fmt) |
| self.warning('class C:\n' |
| ' def __not_special__(self):\n' |
| ' pass\n', |
| 2, ClassChecks.SpecialCheck.notSpecial, |
| '__not_special__') |
| |
| def testUncheckableAttribute(self): |
| # inherit from local variable |
| self.silent('def f(klass):\n' |
| ' class C(klass):\n' |
| ' def __init__(self, x):\n' |
| ' self.foo = x\n' |
| ' def g(self): return self.foo\n' |
| ' return C(1)\n') |
| # inherit from an expression |
| self.silent('class A:\n' |
| ' def __add__(self, unused):\n' |
| ' return A\n' |
| 'class B(A): pass\n' |
| 'class C(A() + B()): pass\n') |
| # inherit from something not in a source module |
| self.warning('import exceptions\n' |
| 'class C(exceptions):\n' |
| ' def f(self):\n' |
| ' return self.value\n', |
| 4, ClassChecks.AttributeCheck.unknownAttribute, |
| 'C', 'value') |
| self.warning('import pychecker2.utest.data\n' |
| 'class C(pychecker2.utest.data.exceptions.AssertionError):\n' |
| ' def f(self):\n' |
| ' return self.value\n', |
| 4, ClassChecks.AttributeCheck.unknownAttribute, |
| 'C', 'value') |
| self.warning('from pychecker2.utest.data import Data as F\n' |
| 'class C(F.DataError):\n' |
| ' def f(self):\n' |
| ' return self.value\n', |
| 4, ClassChecks.AttributeCheck.unknownAttribute, |
| 'C', 'value') |
| self.silent('def f():\n' |
| ' class Foo(None.__class__): pass\n' |
| ' return Foo()\n') |
| |
| def testUnused(self): |
| self.warning('class C:\n' |
| ' def __init__(self):\n' |
| ' self.__x = 1\n', |
| 3, ClassChecks.AttributeCheck.unusedAttribute, '__x', 'C') |
| self.warning('class C:\n' |
| ' def __init__(self):\n' |
| ' class D:\n' |
| ' def __init__(self): self.__x = 1\n' |
| ' def f(self):\n' |
| ' return self.__x\n' |
| ' self.__x = D()\n', |
| 7, ClassChecks.AttributeCheck.unusedAttribute, '__x', 'C') |
| self.silent('class C:\n' |
| ' def __init__(self):\n' |
| ' self.x = 1\n') |
| |
| def testInit(self): |
| w = ClassChecks.InitCheck.initReturnsValue |
| self.warning('class C:\n' |
| ' def __init__(self):\n' |
| ' return 1\n', 3, w) |
| self.warning('class C:\n' |
| ' def __init__(self):\n' |
| ' def f(x):\n' |
| ' return f(x - 1)\n' |
| ' return f(1)\n', 5, w) |
| |
| w = ClassChecks.AttributeCheck.attributeInitialized |
| self.silent('class C:\n' |
| ' def f(self):\n' |
| ' self.value = 1\n' |
| ' def __init__(self):\n' |
| ' self.value = None\n') |
| self.warning('class C:\n' |
| ' def __init__(self): pass\n' |
| ' def f(self):\n' |
| ' self.value = 1', |
| 4, w, 'value') |
| |
| def testRepr(self): |
| w = ClassChecks.ReprCheck.backquoteSelf |
| self.silent('class C:\n' |
| ' def __str__(self):\n' |
| ' return "C" + `self`\n') |
| self.warning('class C:\n' |
| ' def __repr__(self):\n' |
| ' return "C" + `self`\n', 3, w) |
| |
| def testMethodRedefined(self): |
| w = ClassChecks.AttributeCheck.methodRedefined |
| |
| self.warning('class C:\n' |
| ' def f(self):\n' |
| ' self.f = lambda x: x\n', |
| 3, w, 'f', 'C') |
| self.warning('class B:\n' |
| ' def g(self): pass\n' |
| 'class C(B):\n' |
| ' def f(self):\n' |
| ' self.g = lambda x: x\n', |
| 5, w, 'g', 'C') |
| self.warning('from pychecker2.utest.data import Data\n' |
| 'class C(Data):\n' |
| ' def f(self):\n' |
| ' self.get_value = 1\n', |
| 4, w, 'get_value','C') |
| |
| |
