def test_tag_addition_with_single_selection(self): scope = Scope(database=self._database) scope.select_document(7) self.assertEqual(scope.get_concept_tags(), []) document_only_tags = scope.get_selection_only_tags() self.assertEqual(len(document_only_tags), 2) _ = scope.create_tag('script') _ = scope.create_tag('web') selection_only_tags = scope.get_selection_only_tags() self.assertEqual(len(selection_only_tags), 4) for tag in selection_only_tags: self.assertIn(tag.name, ['rust', 'index', 'script', 'web'])
def test_tag_removing_with_selection_from_document(self): scope = Scope(database=self._database) scope.add_tag(2) scope.add_tag(5) scope.select_document(5) concept_tags = scope.get_concept_tags() self.assertEqual(len(concept_tags), 2) for tag in concept_tags: self.assertIn(tag.name, ['gui', 'python']) selection_only_tags = scope.get_selection_only_tags() self.assertEqual(len(selection_only_tags), 1) self.assertEqual(selection_only_tags[0].name, 'book') scope.remove_tag(1) concept_tags = scope.get_concept_tags() self.assertEqual(len(concept_tags), 2) for tag in concept_tags: self.assertIn(tag.name, ['gui', 'python']) self.assertEqual(scope.get_selection_only_tags(), [])
def test_document_deselection(self): scope = Scope(database=self._database) expected_tag_ids = {6: [2], 4: [1, 2], 1: [1, 2, 5]} for document_id in [1, 4, 5, 6]: scope.select_document(document_id) for document_id in [6, 4, 1]: scope.deselect_document(document_id) selection_only_tags = scope.get_selection_only_tags() self.assertEqual(len(selection_only_tags), len(expected_tag_ids[document_id])) for tag in selection_only_tags: self.assertIn(tag.id, expected_tag_ids[document_id])
def test_tag_creation_with_single_document(self): scope = Scope(database=self._database) scope.select_document(4) tag = scope.create_tag('template') scope.add_tag(tag.id) self.assertEqual(scope.get_concept_document_ids(), [4]) self.assertEqual(scope.get_concept_tag_ids(), [tag.id]) self.assertEqual(scope.get_selection_document_ids(), [4]) self.assertEqual(set(scope.get_selection_tag_ids()), {2, 6, tag.id}) self.assertEqual(scope.get_concept_only_document_ids(), []) self.assertEqual(set(scope.get_selection_only_tag_ids()), {2, 6}) tag_names = {2: 'python', 6: 'index', tag.id: 'template'} for tag in scope.get_selection_only_tags(): self.assertEqual(tag.name, tag_names[tag.id])
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])
def test_tag_destroying(self): scope = Scope(database=self._database) tag_counts = { 1: [2, 1, 0, 0, 0, 0, 0], 2: [2, 1, 1, 0, 0, 0, 0], 3: [2, 1, 1, 1, 0, 0, 0], 4: [2, 2, 1, 1, 1, 1, 0], 5: [3, 2, 1, 1, 1, 0, 0], 6: [2, 2, 2, 2, 1, 0, 0], 7: [2, 2, 2, 1, 1, 1, 0], 8: [2, 2, 2, 1, 1, 0, 0] } for i in range(7): if i > 0: scope.destroy_tag(i) for document_id in range(1, 9): scope.select_document(document_id) self.assertEqual(len(scope.get_selection_only_tags()), tag_counts[document_id][i]) scope.deselect_document(document_id)
def test_document_selection(self): scope = Scope(database=self._database) document_selections = { (1, 2): [1], (1, 5): [1, 2], (2, 7): [3], (5, 6, 8): [5], (2, 7, 8): [3], (3, 4): [], (5, 7): [], (1, 2, 3, 4, 5, 6, 7, 8): [] } for selection, tag_ids in document_selections.items(): for document_id in selection: scope.select_document(document_id) selection_only_tags = scope.get_selection_only_tags() self.assertEqual(len(selection_only_tags), len(tag_ids)) for tag in selection_only_tags: self.assertIn(tag.id, tag_ids) for document_id in selection: scope.deselect_document(document_id)
def test_toggle_document_selection(self): scope = Scope(database=self._database) toggle_items = [{ 'document_id': 1, 'tag_ids': [1, 2] }, { 'document_id': 2, 'tag_ids': [1] }, { 'document_id': 1, 'tag_ids': [1, 3] }, { 'document_id': 7, 'tag_ids': [3] }, { 'document_id': 8, 'tag_ids': [3] }, { 'document_id': 6, 'tag_ids': [] }, { 'document_id': 2, 'tag_ids': [] }, { 'document_id': 7, 'tag_ids': [5] }, { 'document_id': 8, 'tag_ids': [4, 5] }] for item in toggle_items: scope.toggle_document_selection(item['document_id']) selection_only_tags = scope.get_selection_only_tags() self.assertEqual(len(selection_only_tags), len(item['tag_ids'])) for tag in selection_only_tags: self.assertIn(tag.id, item['tag_ids'])