def test_givenABPembDownload_whenModelIsNotLocal_thenDownloadEmbeddings(
self, weights_download_mock):
with patch("deepparse.cli.download.CACHE_PATH", self.fake_cache_path):
with patch("deepparse.cli.download.BPEmb") as downloader:
download.main([self.a_bpemb_model_type])
downloader.assert_any_call(lang="multi", vs=100000,
dim=300) # settings for BPemb
def test_givenAFasttextDownload_whenModelIsNotLocal_thenDownloadEmbeddings(
self, weights_download_mock):
with patch("deepparse.cli.download.CACHE_PATH", self.fake_cache_path):
with patch("deepparse.cli.download.download_fasttext_embeddings"
) as downloader:
download.main([self.a_fasttext_model_type])
downloader.assert_any_call(saving_dir=self.fake_cache_path)
def test_givenABPembDownload_whenModelIsNotLocalButNotLatest_thenDownloadWeights(
self, download_embeddings_mock, os_is_file_mock):
with patch("deepparse.cli.download.CACHE_PATH", self.fake_cache_path):
with patch(
"deepparse.cli.download.download_weights") as downloader:
download.main([self.a_bpemb_model_type])
downloader.assert_any_call(self.a_bpemb_model_type,
self.fake_cache_path)
def test_givenAFasttextDownload_whenModelIsLocalAndGoodVersion_thenDoNoting(
self, download_embeddings_mock, os_is_file_mock,
latest_version_mock):
with patch("deepparse.cli.download.CACHE_PATH", self.fake_cache_path):
with patch(
"deepparse.cli.download.download_weights") as downloader:
download.main([self.a_fasttext_model_type])
downloader.assert_not_called()
def test_givenABPembDownload_whenModelIsLocalAndGoodVersion_thenDoNoting(
self, download_embeddings_mock, os_is_file_mock,
latest_version_mock):
os_is_file_mock.return_value = True
latest_version_mock.return_value = True # the latest version
with patch("deepparse.cli.download.CACHE_PATH", self.fake_cache_path):
with patch(
"deepparse.cli.download.download_weights") as downloader:
download.main([self.a_bpemb_model_type])
downloader.assert_not_called()
