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()