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_infer_client_method(): assert known_types_for_module("""\ import boto3 a = boto3.client('ec2').describe_instances """) == { 'boto3': Boto3ModuleType(), 'a': Boto3ClientMethodType('ec2', 'describe_instances') }