| from pathlib import Path |
|
|
| import fitz |
| from ai_module.nlp.profile_generator import ProfileGenerator |
| from app.services.cv_extractor import extract_text_from_pdf, save_text_as_txt |
|
|
|
|
| def test_generate_profile_from_job_description(): |
| text = ( |
| "Senior Python developer with 5 years of experience in FastAPI, Docker, and SQL. " |
| "Strong communication and teamwork skills are required. " |
| "Bachelor degree in Computer Science. Fluent in English and French." |
| ) |
|
|
| profile = ProfileGenerator.generate_from_text(text) |
|
|
| assert profile["ideal_experience_years"] >= 5 |
| assert any(skill["name"].lower() == "python" for skill in profile["ideal_skills"]) |
| assert any(skill["name"].lower() == "fastapi" for skill in profile["ideal_skills"]) |
| assert "Bachelor" in profile["ideal_education"] |
| assert "English" in profile["ideal_languages"] |
| assert "French" in profile["ideal_languages"] |
|
|
|
|
| def test_extract_text_from_pdf_and_save_txt(tmp_path: Path): |
| pdf_path = tmp_path / "sample_cv.pdf" |
| doc = fitz.open() |
| page = doc.new_page() |
| page.insert_text((72, 72), "Hello world from CV extraction test") |
| doc.save(str(pdf_path)) |
| doc.close() |
|
|
| extracted_text = extract_text_from_pdf(str(pdf_path)) |
| assert "Hello world from CV extraction test" in extracted_text |
|
|
| txt_dir = tmp_path / "txt" |
| txt_path = save_text_as_txt(extracted_text, str(txt_dir), pdf_path.name) |
| assert Path(txt_path).exists() |
| assert Path(txt_path).read_text(encoding="utf-8").strip() == extracted_text.strip() |
|
|
