def test_type_equality():
assert Boto3ModuleType() == Boto3ModuleType()
assert Boto3CreateClientType() == Boto3CreateClientType()
assert Boto3ModuleType() != Boto3CreateClientType()
assert Boto3ClientType('s3') == Boto3ClientType('s3')
assert Boto3ClientType('s3') != Boto3ClientType('ec2')
assert Boto3ClientType('s3') == Boto3ClientType('s3')
assert (Boto3ClientMethodType('s3', 'list_objects') ==
Boto3ClientMethodType('s3', 'list_objects'))
assert (Boto3ClientMethodType('ec2', 'describe_instances') !=
Boto3ClientMethodType('s3', 'list_object'))
assert (Boto3ClientMethodType('ec2', 'describe_instances') !=
Boto3CreateClientType())
def test_inferred_client_create_type():
assert known_types_for_module("""\
import boto3
a = boto3.client
""") == {
'boto3': Boto3ModuleType(),
'a': Boto3CreateClientType()
}
def test_inferred_client_type_each_part():
assert known_types_for_module("""\
import boto3
a = boto3.client
b = a('ec2')
""") == {'boto3': Boto3ModuleType(),
'a': Boto3CreateClientType(),
'b': Boto3ClientType('ec2')}