Ejemplo n.º 1
0
def test__similar_users__bad():
    """ensure the similar_users() method raises an exception for an user not in training data"""

    with pytest.raises(AssertionError):
        model = RankFM(factors=2)
        model.fit(intx_train_pd_int)
        similar = model.similar_users(9, n_users=1)
Ejemplo n.º 2
0
def test__similar_users__good():
    """test the similar_users() method for a valid [user_id]"""

    model = RankFM(factors=2)
    model.fit(intx_train_pd_int)
    similar = model.similar_users(1, n_users=2)

    shape = similar.shape == (2, )
    users = np.isin(similar, intx_train_pd_int['user_id'].unique()).all()
    assert shape and users