def test_select_documents_with_max_num_documents(self):
all_documents = mock_news_and_comments()
expected_content = [doc.content for doc in all_documents]
documents_content = select_documents(
document_types="both", max_num_documents=example_max_documents)
self.assertEqual(len(documents_content), example_max_documents)
self.assertEqual(documents_content,
expected_content[:example_max_documents])
def test_select_documents(self):
all_documents = mock_news_and_comments()
expected_content = [doc.content for doc in all_documents]
documents_content = select_documents(document_types="both",
max_num_documents=100)
self.assertEqual(documents_content, expected_content)
def test_get_documents_content(self):
all_documents = mock_news_and_comments()
expected_content = [doc.content for doc in all_documents]
documents_list = Document.objects.all()
documents_content = get_documents_content(documents_list)
self.assertEqual(documents_content, expected_content)
def test_select_documents_from_database_both(self):
document_types = "both"
expected_documents = mock_news_and_comments()
selected_documents = select_documents_from_database(document_types)
for i in range(len(expected_documents)):
self.assertEqual(selected_documents[i], expected_documents[i])
def test_select_documents_from_database_comments(self):
document_types = "comments"
mock_news_and_comments()
selected_documents = select_documents_from_database(document_types)
validate_documents(self, selected_documents, comments_content)
def test_select_documents_from_database_news(self):
document_types = "news"
mock_news_and_comments()
expected_content = [news_content[0], news_content[1]]
documents_list = select_documents_from_database(document_types)
validate_documents(self, documents_list, expected_content)