def test_get_current_properties_with_properties_then_last_values(): properties = { "face_color": np.array(["cyan", "red", "red"]), "angle": np.array([0.5, 1.5, 1.5]), } current_properties = get_current_properties(properties, {}, 3) assert current_properties == { "face_color": "red", "angle": 1.5, }
def test_get_current_properties_with_property_choices_then_first_values(): properties = { "face_color": np.empty(0, dtype=str), "angle": np.empty(0, dtype=float), } property_choices = { "face_color": np.array(["cyan", "red"]), "angle": np.array([0.5, 1.5]), } current_properties = get_current_properties(properties, property_choices, 0) assert current_properties == { "face_color": "cyan", "angle": 0.5, }