@@ -154,19 +154,26 @@ def test_calculate_entropy_no_exception(path: Path, draw_plot: bool):
154154
155155
156156@pytest .mark .parametrize (
157- "extract_root, path, extract_dir_prefix " ,
157+ "extract_root, path, result " ,
158158 [
159- ("/extract" , "firmware" , "firmware" ),
160- ("/extract" , "relative/firmware" , "firmware" ),
161- ("/extract" , "/extract/dir/firmware" , "dir/firmware" ),
162- ("/extract/dir" , "/extract/dir/firmware" , "firmware" ),
163- ("/extract" , "/some/place/else/firmware" , "firmware" ),
159+ ("/extract" , "firmware" , "/extract/firmware_extract" ),
160+ ("/extract" , "relative/firmware" , "/extract/firmware_extract" ),
161+ ("/extract" , "/extract/dir/firmware" , "/extract/dir/firmware_extract" ),
162+ (
163+ "/extract/dir" ,
164+ "/extract/dir/firmware" ,
165+ "/extract/dir/firmware_extract" ,
166+ ),
167+ ("/extract" , "/some/place/else/firmware" , "/extract/firmware_extract" ),
168+ (
169+ "extract" ,
170+ "/some/place/else/firmware" ,
171+ str (Path ("." ).resolve () / "extract/firmware_extract" ),
172+ ),
164173 ],
165174)
166175def test_ExtractionConfig_get_extract_dir_for (
167- extract_root : str , path : str , extract_dir_prefix : str
176+ extract_root : str , path : str , result : str
168177):
169178 cfg = ExtractionConfig (extract_root = Path (extract_root ), entropy_depth = 0 )
170- assert cfg .get_extract_dir_for (Path (path )) == (
171- cfg .extract_root / Path (extract_dir_prefix + cfg .extract_suffix )
172- )
179+ assert cfg .get_extract_dir_for (Path (path )) == Path (result )
0 commit comments