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)