modify tests/test_parse.py

sayat-a · Jul 30, 2024 · 1784fdf · 1784fdf
1 parent 61df027
commit 1784fdf
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_parse.py b/tests/test_parse.py
@@ -1,13 +1,15 @@
 from gendiff.parse import load_file, define_extension
 import json
 import yaml
+import pytest
 
 
-def test_define_extension():
-    file_path = 'tests/fixtures/file1.json'
-    file_path2 = 'tests/fixtures/plain_yml_file1.yaml'
-    assert define_extension(file_path) == '.json'
-    assert define_extension(file_path2) == '.yaml'
+@pytest.mark.parametrize("file_path, expected_result", [
+    ('tests/fixtures/file1.json', '.json'),
+    ('tests/fixtures/plain_yml_file1.yaml', '.yaml')
+])
+def test_define_extension(file_path, expected_result):
+    assert define_extension(file_path) == expected_result
 
 
 def test_load_file():