Skip to content

Commit

Permalink
Fixed tests
Browse files Browse the repository at this point in the history
  • Loading branch information
kian79 committed Jan 31, 2024
1 parent be5207b commit 5f3ba7b
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 8 deletions.
9 changes: 5 additions & 4 deletions pdf_processing/test_pdf_extractor.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
from pdf_processing.preprocessing import convert_pdf_to_text

input_pdf_path = 'pdf_processing/sample_pdf.pdf'
result_text = convert_pdf_to_text(input_pdf_path)
expected_text = """Love encompasses a range of strong and positive emotional and mental states, from the most sublime virtue or good habit, the deepest interpersonal affection, to the simplest pleasure.[1] An example of this range of meanings is that the love of a mother differs from the love of a spouse, which differs from the love for food. Most commonly , love refers to a feeling of strong attraction and emotional attachment.[2] Love is considered to be both positive and negative, with its virtue representing human kindness, compassion, and affection—"the unselfish, loyal and benevolent concern for the good of another"—and its vice representing a human moral flaw akin to vanity , selfishness, amour-propre, and egotism, potentially leading people into a type of mania, obsessiveness, or codependency .[3] It may also describe compassionate and affectionate actions towards other humans, oneself, or animals.[4] In its various forms, love acts as a major facilitator of interpersonal relationships and, owing to its central psychological importance, is one of the most common themes in the creative arts.[5][6] Love has been postulated to be a function that keeps human beings together against menaces and to facilitate the continuation of the species.[7]"""
def test_text_extractor():
input_pdf_path = 'sample_pdf.pdf'
result_text = convert_pdf_to_text(input_pdf_path)
expected_text = """Love encompasses a range of strong and positive emotional and mental states, from the most sublime virtue or good habit, the deepest interpersonal affection, to the simplest pleasure.[1] An example of this range of meanings is that the love of a mother differs from the love of a spouse, which differs from the love for food. Most commonly , love refers to a feeling of strong attraction and emotional attachment.[2] Love is considered to be both positive and negative, with its virtue representing human kindness, compassion, and affection—"the unselfish, loyal and benevolent concern for the good of another"—and its vice representing a human moral flaw akin to vanity , selfishness, amour-propre, and egotism, potentially leading people into a type of mania, obsessiveness, or codependency .[3] It may also describe compassionate and affectionate actions towards other humans, oneself, or animals.[4] In its various forms, love acts as a major facilitator of interpersonal relationships and, owing to its central psychological importance, is one of the most common themes in the creative arts.[5][6] Love has been postulated to be a function that keeps human beings together against menaces and to facilitate the continuation of the species.[7]"""

assert result_text == expected_text
assert result_text == expected_text
9 changes: 5 additions & 4 deletions pdf_processing/test_text_summarization.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from pdf_processing.preprocessing import summarize_text

input_text = """Love encompasses a range of strong and positive emotional and mental states, from the most sublime virtue or good habit, the deepest interpersonal affection, to the simplest pleasure.[1] An example of this range of meanings is that the love of a mother differs from the love of a spouse, which differs from the love for food. Most commonly , love refers to a feeling of strong attraction and emotional attachment.[2] Love is considered to be both positive and negative, with its virtue representing human kindness, compassion, and affection—"the unselfish, loyal and benevolent concern for the good of another"—and its vice representing a human moral flaw akin to vanity , selfishness, amour-propre, and egotism, potentially leading people into a type of mania, obsessiveness, or codependency .[3] It may also describe compassionate and affectionate actions towards other humans, oneself, or animals.[4] In its various forms, love acts as a major facilitator of interpersonal relationships and, owing to its central psychological importance, is one of the most common themes in the creative arts.[5][6] Love has been postulated to be a function that keeps human beings together against menaces and to facilitate the continuation of the species.[7]"""
result_summary = summarize_text(input_text)
expected_summary = """Most commonly , love refers to a feeling of strong attraction and emotional attachment. [3] It may also describe compassionate and affectionate actions towards other humans, oneself, or animals. [4] In its various forms, love acts as a major facilitator of interpersonal relationships and, owing to its central psychological importance, is one of the most common themes in the creative arts."""
assert result_summary == expected_summary
def test_text_summarization():
input_text = """Love encompasses a range of strong and positive emotional and mental states, from the most sublime virtue or good habit, the deepest interpersonal affection, to the simplest pleasure.[1] An example of this range of meanings is that the love of a mother differs from the love of a spouse, which differs from the love for food. Most commonly , love refers to a feeling of strong attraction and emotional attachment.[2] Love is considered to be both positive and negative, with its virtue representing human kindness, compassion, and affection—"the unselfish, loyal and benevolent concern for the good of another"—and its vice representing a human moral flaw akin to vanity , selfishness, amour-propre, and egotism, potentially leading people into a type of mania, obsessiveness, or codependency .[3] It may also describe compassionate and affectionate actions towards other humans, oneself, or animals.[4] In its various forms, love acts as a major facilitator of interpersonal relationships and, owing to its central psychological importance, is one of the most common themes in the creative arts.[5][6] Love has been postulated to be a function that keeps human beings together against menaces and to facilitate the continuation of the species.[7]"""
result_summary = summarize_text(input_text)
expected_summary = """Most commonly , love refers to a feeling of strong attraction and emotional attachment. [3] It may also describe compassionate and affectionate actions towards other humans, oneself, or animals. [4] In its various forms, love acts as a major facilitator of interpersonal relationships and, owing to its central psychological importance, is one of the most common themes in the creative arts."""
assert result_summary == expected_summary

0 comments on commit 5f3ba7b

Please sign in to comment.