def test_deselect_all_documents(self):
scope = Scope(database=self._database)
for last_document_id in range(1, 10):
for document_id in range(1, last_document_id):
scope.select_document(document_id)
self.assertEqual(len(scope.get_selection_documents()),
last_document_id - 1)
scope.deselect_all_documents()
self.assertEqual(scope.get_selection_documents(), [])
def test_initial_scope(self):
scope = Scope(database=self._database)
concept_document_ids = scope.get_concept_document_ids()
self.assertEqual(set(concept_document_ids), {1, 2, 3, 4, 5, 6, 7, 8})
concept_documents = scope.get_concept_documents()
for concept_document in concept_documents:
self.assertIn(concept_document.id, sample_documents)
self.assertEqual(concept_document.name,
sample_documents[concept_document.id]['name'])
self.assertEqual(concept_document.type,
sample_documents[concept_document.id]['type'])
self.assertEqual(concept_document.path,
sample_documents[concept_document.id]['path'])
concept_tag_ids = scope.get_concept_tag_ids()
self.assertEqual(concept_tag_ids, [])
concept_tags = scope.get_concept_tags()
self.assertEqual(concept_tags, [])
selection_document_ids = scope.get_selection_document_ids()
self.assertEqual(selection_document_ids, [])
selection_documents = scope.get_selection_documents()
self.assertEqual(selection_documents, [])
selection_tag_ids = scope.get_selection_tag_ids()
self.assertEqual(set(selection_tag_ids), {1, 2, 3, 4, 5, 6})
selection_tags = scope.get_selection_tags()
for selection_tag in selection_tags:
self.assertIn(selection_tag.id, sample_tags)
self.assertEqual(selection_tag.name, sample_tags[selection_tag.id])
concept_only_document_ids = scope.get_concept_only_document_ids()
self.assertEqual(set(concept_only_document_ids),
set(concept_document_ids))
concept_only_documents = scope.get_concept_only_documents()
for concept_only_document in concept_only_documents:
self.assertIn(concept_only_document.id, sample_documents)
self.assertEqual(
concept_only_document.name,
sample_documents[concept_only_document.id]['name'])
self.assertEqual(
concept_only_document.type,
sample_documents[concept_only_document.id]['type'])
self.assertEqual(
concept_only_document.path,
sample_documents[concept_only_document.id]['path'])
selection_only_tag_ids = scope.get_selection_only_tag_ids()
self.assertEqual(set(selection_only_tag_ids), {1, 2, 3, 4, 5, 6})
selection_only_tags = scope.get_selection_only_tags()
for selection_only_tag in selection_only_tags:
self.assertIn(selection_only_tag.id, sample_tags)
self.assertEqual(selection_only_tag.name,
sample_tags[selection_only_tag.id])
