From 4511a582051b107f5729b4e4e53098d4060529e4 Mon Sep 17 00:00:00 2001 From: Ethan Coon Date: Fri, 17 May 2024 13:57:12 -0600 Subject: [PATCH] updates unit tests --- .../sources/test/test_file_manager_nhdplus.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/watershed_workflow/sources/test/test_file_manager_nhdplus.py b/watershed_workflow/sources/test/test_file_manager_nhdplus.py index cc1f6131..06e6f4a7 100644 --- a/watershed_workflow/sources/test/test_file_manager_nhdplus.py +++ b/watershed_workflow/sources/test/test_file_manager_nhdplus.py @@ -25,23 +25,20 @@ def nhd(): # having some robustness issues, lets just test a bunch def test_nhdplus_url1(nhd): url = nhd._url('0204') - assert ( - 'https://prd-tnm.s3.amazonaws.com/StagedProducts/Hydrography/NHDPlusHR/Beta/GDB/NHDPLUS_H_0204_HU4_GDB.zip' - == url) + assert (url.endswith('_GDB.zip')) + assert (url.split('/')[-1].startswith('NHDPLUS_H_0204_HU4')) def test_nhdplus_url2(nhd): url = nhd._url('0601') - assert ( - 'https://prd-tnm.s3.amazonaws.com/StagedProducts/Hydrography/NHDPlusHR/Beta/GDB/NHDPLUS_H_0601_HU4_GDB.zip' - == url) + assert (url.endswith('_GDB.zip')) + assert (url.split('/')[-1].startswith('NHDPLUS_H_0601_HU4')) def test_nhdplus_url3(nhd): url = nhd._url('1402') - assert ( - 'https://prd-tnm.s3.amazonaws.com/StagedProducts/Hydrography/NHDPlusHR/Beta/GDB/NHDPLUS_H_1402_HU4_GDB.zip' - == url) + assert (url.endswith('_GDB.zip')) + assert (url.split('/')[-1].startswith('NHDPLUS_H_1402_HU4')) def test_nhdplus_url_fail(nhd): @@ -86,7 +83,7 @@ def test_nhdplus12(nhd): # download hydrography profile, huc = nhd.get_huc('060102020103') profile, rivers = nhd.get_hydro('060102020103') - assert (202 == len(rivers)) + assert (198 == len(rivers)) def test_vaa(nhd):