from wmera.utils import rel_path_to_file
__author__ = 'Dani'
from test.t_utils.t_factory import get_clean_repo_counter_memory, get_clean_repo_song_memory, \
get_clean_repo_artist_memory, get_clean_repo_artist_mongo, get_clean_repo_counter_mongo, \
get_clean_repo_songs_mongo
from wmera.adapters.in_memory_to_mogno import dump_in_memory_ngrams_into_mongo_ngrams, \
dump_in_memory_counter_into_mongo_counter
# Artist ngrmas
mongo_artist = get_clean_repo_artist_mongo()
memory_artist = get_clean_repo_artist_memory()
memory_artist.load_content(rel_path_to_file("../../files/out/artist_ngrams_usos.json",
__file__))
dump_in_memory_ngrams_into_mongo_ngrams(in_memory_repo=memory_artist,
mongo_repo=mongo_artist)
memory_artist = None # Free memory
# Song ngrams
mongo_song = get_clean_repo_songs_mongo()
memory_song = get_clean_repo_song_memory()
memory_song.load_content(rel_path_to_file("../../files/out/song_ngrams_usos.json",
__file__))
dump_in_memory_ngrams_into_mongo_ngrams(in_memory_repo=memory_song,
mongo_repo=mongo_song)
memory_song = None # Free memory
def test_ngrams_in_memory(self):
self._battery_with_a_concrete_repo(get_clean_repo_artist_memory())
