def test_infer_unary_op_sig(self) -> None:
for op in ('neg', 'pos'):
assert_equal(infer_method_sig('__%s__' % op), '()')
def test_infer_setitem_sig(self) -> None:
assert_equal(infer_method_sig('__setitem__'), '(index, object)')
def test_infer_binary_op_sig(self) -> None:
for op in ('eq', 'ne', 'lt', 'le', 'gt', 'ge',
'add', 'radd', 'sub', 'rsub', 'mul', 'rmul'):
assert_equal(infer_method_sig('__%s__' % op), '(other)')
def test_infer_hash_sig(self) -> None:
assert_equal(infer_method_sig('__hash__'), '()')
def test_infer_getitem_sig(self) -> None:
assert_equal(infer_method_sig('__getitem__'), '(index)')
def test_infer_binary_op_sig(self) -> None:
for op in ('eq', 'ne', 'lt', 'le', 'gt', 'ge', 'add', 'radd', 'sub',
'rsub', 'mul', 'rmul'):
assert_equal(infer_method_sig('__%s__' % op),
[self_arg, ArgSig(name='other')])
def test_infer_setitem_sig(self) -> None:
assert_equal(
infer_method_sig('__setitem__'),
[self_arg, ArgSig(name='index'),
ArgSig(name='object')])
def test_infer_getitem_sig(self) -> None:
assert_equal(infer_method_sig('__getitem__'), [ArgSig(name='index')])
def test_infer_unary_op_sig(self) -> None:
for op in ('neg', 'pos'):
assert_equal(infer_method_sig(f'__{op}__'), [self_arg])
def test_infer_setitem_sig(self) -> None:
assert_equal(infer_method_sig('__setitem__'),
[ArgSig(name='index'), ArgSig(name='object')])
