def test_get_models_remove_duplicates():
"""Assert that duplicate models are returned."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.run(["LR1", "LR2"])
assert atom._get_models(["LR1", "LR1"]) == ["LR1"]
def test_get_models_str():
"""Assert that the right model is returned when the parameter is a string."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.run(["LR1", "LR2"])
assert atom._get_models("Lr1") == ["LR1"]
def test_get_models_list():
"""Assert that the right models are returned when the parameter is a list."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.run(["LR1", "LR2", "LR3"])
assert atom._get_models(["LR1", "LR2"]) == ["LR1", "LR2"]
def test_get_models_empty():
"""Assert that all models are returned when the parameter is empty."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.run(["LR1", "LR2"])
assert atom._get_models(None) == ["LR1", "LR2"]