def test_retrieve_untrained_definition_list():
definition_list = [{
'type': 'a'
}, {
'type': 'b',
'untrainedCategory': True
}, {
'type': 'c',
'untrainedColor': True
}, {
'type': 'd',
'untrainedCombination': True
}, {
'type': 'e',
'untrainedShape': True
}, {
'type': 'f',
'untrainedSize': True
}, {
'type': 'g',
'untrainedCategory': True,
'untrainedColor': True,
'untrainedCombination': True,
'untrainedShape': True,
'untrainedSize': True
}, {
'type': 'h'
}]
actual_1 = retrieve_untrained_definition_list([definition_list],
'untrainedShape')[0]
assert len(actual_1) == 1
assert actual_1[0]['type'] == 'e'
actual_2 = retrieve_untrained_definition_list([definition_list],
'untrainedSize')[0]
assert len(actual_2) == 1
assert actual_2[0]['type'] == 'f'
def test_intuitive_physics_move_across_all_objects_untrained_shapes():
definition_list = util.retrieve_complete_definition_list(
[objects.get_intuitive_physics(False)]
)[0]
trained_list = util.retrieve_trained_definition_list([definition_list])[0]
untrained_list = util.retrieve_untrained_definition_list(
[definition_list],
tags.SCENE.UNTRAINED_SHAPE
)[0]
for definition_1 in trained_list:
option_list = [
definition_2 for definition_2 in untrained_list
if util.is_similar_except_in_shape(definition_1, definition_2,
True)
]
assert len(option_list) >= 2
def test_intuitive_physics_fall_down_complex_objects_untrained_sizes():
definition_list = util.retrieve_complete_definition_list(
[objects.get_intuitive_physics(True, False, True)]
)[0]
trained_list = util.retrieve_trained_definition_list([definition_list])[0]
untrained_list = util.retrieve_untrained_definition_list(
[definition_list],
tags.SCENE.UNTRAINED_SIZE
)[0]
for definition_1 in trained_list:
option_list = [
definition_2 for definition_2 in untrained_list
if util.is_similar_except_in_size(definition_1, definition_2,
True)
]
# We want at least two possible untrained objects.
assert len(option_list) >= 2
def test_retrieve_untrained_definition_list_occluder_objects():
assert len(retrieve_untrained_definition_list(
objects.get(objects.ObjectDefinitionList.OCCLUDERS),
'untrainedShape'
)) > 0
def test_retrieve_untrained_definition_list_obstacle_objects():
assert len(retrieve_untrained_definition_list(
objects.get(objects.ObjectDefinitionList.OBSTACLES),
'untrainedShape'
)) > 0
def test_retrieve_untrained_definition_list_container_objects():
assert len(retrieve_untrained_definition_list(
objects.get(objects.ObjectDefinitionList.CONTAINERS),
'untrainedShape'
)) > 0
def test_retrieve_untrained_definition_list_all_objects():
assert len(retrieve_untrained_definition_list(
objects.get(objects.ObjectDefinitionList.ALL),
'untrainedShape'
)) > 0
