def test_download_dastcom5_creates_folder( mock_isdir, mock_zipfile, mock_makedirs, mock_request ): mock_isdir.return_value = False mock_zipfile.is_zipfile.return_value = False dastcom5.download_dastcom5() mock_makedirs.assert_called_once_with(dastcom5.POLIASTRO_LOCAL_PATH)
def test_download_dastcom5_downloads_file(mock_request, mock_isdir, mock_zipfile): mock_isdir.side_effect = lambda x: x == dastcom5.POLIASTRO_LOCAL_PATH mock_zipfile.is_zipfile.return_value = False dastcom5.download_dastcom5() mock_request.assert_called_once_with(dastcom5.FTP_DB_URL + 'dastcom5.zip', os.path.join(dastcom5.POLIASTRO_LOCAL_PATH, 'dastcom5.zip'), dastcom5._show_download_progress)
def test_download_dastcom5_downloads_file(mock_request, mock_isdir, mock_zipfile): mock_isdir.side_effect = lambda x: x == dastcom5.POLIASTRO_LOCAL_PATH mock_zipfile.is_zipfile.return_value = False dastcom5.download_dastcom5() mock_request.assert_called_once_with( dastcom5.FTP_DB_URL + "dastcom5.zip", os.path.join(dastcom5.POLIASTRO_LOCAL_PATH, "dastcom5.zip"), dastcom5._show_download_progress, )
def test_download_dastcom5_raises_error_when_folder_exists( mock_request, mock_isdir, mock_zipfile, mock_makedirs ): mock_isdir.side_effect = lambda x: x == os.path.join( dastcom5.POLIASTRO_LOCAL_PATH, "dastcom5" ) with pytest.raises(FileExistsError): dastcom5.download_dastcom5() mock_isdir.assert_called_once_with( os.path.join(dastcom5.POLIASTRO_LOCAL_PATH, "dastcom5") )
def main(): parser = argparse.ArgumentParser( prog="poliastro", description="Command line tools for the poliastro Python library.") parser.add_argument("--version", action='version', version=poliastro.__version__) parser.add_argument("--download-dastcom5", action="store_true", help="Downloads DASTCOM5 database") args = parser.parse_args() if args.download_dastcom5: download_dastcom5() else: parser.print_help()
def main(): parser = argparse.ArgumentParser( prog="poliastro", description="Command line tools for the poliastro Python library.") parser.add_argument("--version", action='version', version=poliastro.__version__) parser.add_argument("--download-dastcom5", action="store_true", help="Downloads DASTCOM5 database") args = parser.parse_args() if args.download_dastcom5: download_dastcom5() else: parser.print_help()
def test_download_dastcom5_creates_folder(mock_isdir, mock_zipfile, mock_makedirs, mock_request): mock_isdir.return_value = False mock_zipfile.is_zipfile.return_value = False dastcom5.download_dastcom5() mock_makedirs.assert_called_once_with(dastcom5.POLIASTRO_LOCAL_PATH)