Spaces:
Sleeping
Sleeping
File size: 1,543 Bytes
9df97a2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | 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()
|