test fix

Author: khairahscorner

tools/tests/automation_functional_test.py

@@ -20,10 +20,9 @@ def test_write_mentors_skip_zero_rows(monkeypatch):
     run_automation()
 
     result = read_yml_file(tmp_filename)
-    assert len(result) == 3, f"Expected to write 3 mentors but added {len(result)}"
+    assert len(result) == 2, f"Expected to write 2 mentors but added {len(result)}"
     assert MENTOR_2 == result[0]['name'], f"Expected content to be {MENTOR_2} but got '{result[0]['name']}'"
     assert MENTOR_3 == result[1]['name'], f"Expected content to be {MENTOR_3} but got '{result[1]['name']}'"
-    assert MENTOR_4 == result[2]['name'], f"Expected content to be {MENTOR_4} but got '{result[2]['name']}'"
 
     # Clean up the temporary file
     os.remove(tmp_filename)