from cbt_knowledge import find_similar_situations


def test_find_similar_situations_returns_shape():
    items = find_similar_situations("AO", num_situations=2)
    assert isinstance(items, list) and len(items) == 2
    for it in items:
        assert set(["situation", "distorted", "reframed"]).issubset(it.keys())