def test_prepare_properties_with_properties_and_choices_then_merge_choice_values(
):
properties, choices = prepare_properties({"aa": [1, 3]}, {"aa": [1, 2]},
num_data=2)
assert list(properties.keys()) == ["aa"]
assert np.array_equal(properties["aa"], [1, 3])
assert list(choices.keys()) == ["aa"]
assert np.array_equal(choices["aa"], [1, 2, 3])
def test_prepare_properties_with_properties_and_choices_and_save_choices():
properties, choices = prepare_properties(
{"aa": [1, 3]},
{
"aa": [1, 2],
"bb": [7, 6]
},
num_data=2,
save_choices=True,
)
assert list(properties.keys()) == ["aa", "bb"]
assert np.array_equal(properties["aa"], [1, 3])
assert np.array_equal(properties["bb"], [None, None])
assert list(choices.keys()) == ["aa", "bb"]
assert np.array_equal(choices["aa"], [1, 2, 3])
assert np.array_equal(choices["bb"], [6, 7])
def test_prepare_properties_with_choices_then_properties_are_none():
properties, choices = prepare_properties({}, {"aa": [1, 2]}, num_data=2)
assert list(properties.keys()) == ["aa"]
assert np.array_equal(properties["aa"], [None, None])
assert list(choices.keys()) == ["aa"]
assert np.array_equal(choices["aa"], [1, 2])
def test_prepare_properties_with_properties_and_choices():
properties, choices = prepare_properties({"aa": [1, 2, 1]}, num_data=3)
assert np.array_equal(properties["aa"], [1, 2, 1])
assert np.array_equal(choices["aa"], [1, 2])
def test_prepare_properties_with_empty_properties_and_choices():
assert prepare_properties({}, {}) == ({}, {})
