Skip to content

Commit e823d3e

Browse files
committed
added short tests for requirements.txt
Signed-off-by: Samuel Babak <samuel.babak@sas.com>
1 parent 55251eb commit e823d3e

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tests/unit/test_write_json_files.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -701,6 +701,7 @@ def test_create_requirements_json(change_dir):
701701
pickle.dump(dtc, pkl_file)
702702
jf.create_requirements_json(tmp_dir, Path(tmp_dir), True)
703703
assert (Path(tmp_dir) / "requirements.json").exists()
704+
assert (Path(tmp_dir) / "requirements.txt").exists()
704705

705706
json_dict = jf.create_requirements_json(tmp_dir)
706707
expected = [
@@ -709,13 +710,20 @@ def test_create_requirements_json(change_dir):
709710
"command": f"pip install numpy=={np.__version__}",
710711
},
711712
{
712-
"step": "install sklearn",
713-
"command": f"pip install sklearn=={sk.__version__}",
713+
"step": "install scikit-learn",
714+
"command": f"pip install scikit-learn=={sk.__version__}",
714715
},
715716
]
716717
unittest.TestCase.maxDiff = None
717718
unittest.TestCase().assertCountEqual(json_dict, expected)
718719

720+
# Verify requirements.txt content
721+
with open(Path(tmp_dir) / "requirements.txt", "r") as file:
722+
requirements_content = [line.strip() for line in file.readlines()]
723+
724+
assert f"numpy=={np.__version__}" in requirements_content
725+
assert f"scikit-learn=={sk.__version__}" in requirements_content
726+
719727

720728
class TestAssessBiasHelpers(unittest.TestCase):
721729
md_1 = pd.DataFrame({"Value": [0], "Base": ["A"], "Compare": ["C"]})

0 commit comments

Comments
 (0)