67 lines
1.4 KiB
Python
67 lines
1.4 KiB
Python
|
from IPython.utils.dir2 import dir2
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
class Base(object):
|
||
|
|
x = 1
|
||
|
|
z = 23
|
||
|
|
|
||
|
|
|
||
|
|
def test_base():
|
||
|
|
res = dir2(Base())
|
||
|
|
assert "x" in res
|
||
|
|
assert "z" in res
|
||
|
|
assert "y" not in res
|
||
|
|
assert "__class__" in res
|
||
|
|
assert res.count("x") == 1
|
||
|
|
assert res.count("__class__") == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_SubClass():
|
||
|
|
class SubClass(Base):
|
||
|
|
y = 2
|
||
|
|
|
||
|
|
res = dir2(SubClass())
|
||
|
|
assert "y" in res
|
||
|
|
assert res.count("y") == 1
|
||
|
|
assert res.count("x") == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_SubClass_with_trait_names_attr():
|
||
|
|
# usecase: trait_names is used in a class describing psychological classification
|
||
|
|
|
||
|
|
class SubClass(Base):
|
||
|
|
y = 2
|
||
|
|
trait_names = 44
|
||
|
|
|
||
|
|
res = dir2(SubClass())
|
||
|
|
assert "trait_names" in res
|
||
|
|
|
||
|
|
|
||
|
|
def test_misbehaving_object_without_trait_names():
|
||
|
|
# dir2 shouldn't raise even when objects are dumb and raise
|
||
|
|
# something other than AttribteErrors on bad getattr.
|
||
|
|
|
||
|
|
class MisbehavingGetattr:
|
||
|
|
def __getattr__(self, attr):
|
||
|
|
raise KeyError("I should be caught")
|
||
|
|
|
||
|
|
def some_method(self):
|
||
|
|
return True
|
||
|
|
|
||
|
|
class SillierWithDir(MisbehavingGetattr):
|
||
|
|
def __dir__(self):
|
||
|
|
return ["some_method"]
|
||
|
|
|
||
|
|
for bad_klass in (MisbehavingGetattr, SillierWithDir):
|
||
|
|
obj = bad_klass()
|
||
|
|
|
||
|
|
assert obj.some_method()
|
||
|
|
|
||
|
|
with pytest.raises(KeyError):
|
||
|
|
obj.other_method()
|
||
|
|
|
||
|
|
res = dir2(obj)
|
||
|
|
assert "some_method" in res
|