def test_reading():
    """
    Test reading of the movielens dataset
    """
    training_file = "training.pt"
    test_file = "test.pt"
    processed_folder = f"{MOVIELENS_ROOT}/MovieLens/processed"

    assert torch.load(os.path.join(processed_folder, test_file)).size() == torch.Size(
        [943, 1682]
    )
    assert torch.load(
        os.path.join(processed_folder, training_file)
    ).size() == torch.Size([943, 1682])

    Test reading of the movielens dataset
    
tests/contrib/datasets/test_movielens.py