@@ -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
720728class TestAssessBiasHelpers (unittest .TestCase ):
721729 md_1 = pd .DataFrame ({"Value" : [0 ], "Base" : ["A" ], "Compare" : ["C" ]})
0 commit comments