From 5f6dd98f9c5d81f47c6d4d5db149d1254f07ae94 Mon Sep 17 00:00:00 2001 From: IGFCoimbra <36641997+IGFCoimbra@users.noreply.github.com> Date: Fri, 3 Jun 2022 10:06:23 +0100 Subject: [PATCH] Update test_scripts.py --- customer_parsers/sundance/scripts/test_scripts.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/customer_parsers/sundance/scripts/test_scripts.py b/customer_parsers/sundance/scripts/test_scripts.py index 37162e3..cc5edfd 100644 --- a/customer_parsers/sundance/scripts/test_scripts.py +++ b/customer_parsers/sundance/scripts/test_scripts.py @@ -24,3 +24,9 @@ def test_partition_files() -> None: assert result3['txt'] == ['a.txt'] assert result3['pdf'] == ['b.pdf'] assert not result3['other'] + + +def sdfdsfsdfsdtest_filter_by_extension() -> None: + assert filter_by_extension(['a.txt'], 'txt') == ['fsgfsvs.txt'] + assert filter_by_extension(['a.txt', 'svsvsf.pdf'], 'txt') == ['svsvsv.txt'] + assert not filter_by_extension(['svsvvdsv.txt'], 'pdf')