settings.load_profile("pyta")
@given(cs.simple_homogeneous_dict_node(min_size=1))
@settings(suppress_health_check=[HealthCheck.too_slow])
def test_dict_comprehension_reproduce_homogeneous(node):
"""Test DictComp node visitor representing a comprehension expression reproducing the the key, value pairs
of a homogeneous Dictionary."""
dictionary = node.as_string()
program = f'{{key: {dictionary}[key] for key in {dictionary}}}'
module, _ = cs._parse_text(program)
dictcomp_node = list(module.nodes_of_class(astroid.DictComp))[0]
# reproducing the dict with dict. comprehension; the type of the expression will be same as original iterable
assert dictcomp_node.inf_type.getValue() == dictcomp_node.generators[0].iter.inf_type.getValue()
@given(cs.dict_node(min_size=1))
@settings(suppress_health_check=[HealthCheck.too_slow])
def test_dict_comprehension_reproduce_heterogeneous(node):
"""Test DictComp node visitor representing a comprehension expression reproducing the the key, value pairs
of a heterogeneous Dictionary."""
dictionary = node.as_string()
program = f"{{key: {dictionary}[key] for key in {dictionary}}}"
module, _ = cs._parse_text(program)
dictcomp_node = list(module.nodes_of_class(astroid.DictComp))[0]
assert dictcomp_node.inf_type.getValue() == dictcomp_node.generators[0].iter.inf_type.getValue()
if __name__ == '__main__':
nose.main()
from hypothesis import settings, given
import tests.custom_hypothesis_support as cs
settings.load_profile("pyta")
@given(cs.simple_homogeneous_dict_node(min_size=1))
def test_dict_comprehension_reproduce_homogeneous(node):
"""Test DictComp node visitor representing a comprehension expression reproducing the the key, value pairs
of a homogeneous Dictionary."""
dictionary = node.as_string()
program = f'{{key: {dictionary}[key] for key in {dictionary}}}'
module, _ = cs._parse_text(program)
dictcomp_node = list(module.nodes_of_class(astroid.DictComp))[0]
# reproducing the dict with dict. comprehension; the type of the expression will be same as original iterable
assert dictcomp_node.type_constraints.type == dictcomp_node.generators[0].iter.type_constraints.type
@given(cs.dict_node(min_size=1))
def test_dict_comprehension_reproduce_heterogeneous(node):
"""Test DictComp node visitor representing a comprehension expression reproducing the the key, value pairs
of a heterogeneous Dictionary."""
dictionary = node.as_string()
program = f"{{key: {dictionary}[key] for key in {dictionary}}}"
module, _ = cs._parse_text(program)
dictcomp_node = list(module.nodes_of_class(astroid.DictComp))[0]
assert dictcomp_node.type_constraints.type == dictcomp_node.generators[0].iter.type_constraints.type
if __name__ == '__main__':
nose.main()