예제 #1
0
    def test_preprocessing(self):
        # For the most part the necessary checks are performed within
        # _filter_index_sort()

        cache_path = os.path.join(self.temp_data_dir, "test_cache.pickle")
        data, valid_cache = data_preprocessing._filter_index_sort(
            self.rating_file, cache_path=cache_path)

        assert len(data[rconst.USER_MAP]) == NUM_USERS
        assert len(data[rconst.ITEM_MAP]) == NUM_ITEMS
예제 #2
0
  def test_preprocessing(self):
    # For the most part the necessary checks are performed within
    # _filter_index_sort()

    cache_path = os.path.join(self.temp_data_dir, "test_cache.pickle")
    data, valid_cache = data_preprocessing._filter_index_sort(
        self.rating_file, cache_path=cache_path)

    assert len(data[rconst.USER_MAP]) == NUM_USERS
    assert len(data[rconst.ITEM_MAP]) == NUM_ITEMS