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({}, {}) == ({}, {})