def test_get_documents_with_multiple_tags(self): scope = Scope(database=self._database) tag_selections = { (1, 2): [1, 5], (1, 3): [2], (1, 4): [3], (1, 5): [5], (1, 6): [], (2, 5): [5], (3, 6): [7], (1, 2, 5): [5], (3, 5, 6): [], (1, 2, 3, 4, 5, 6): [] } for selection, document_ids in tag_selections.items(): self.assertEqual(scope.get_concept_tags(), []) for tag_id in selection: scope.add_tag(tag_id) documents = scope.get_concept_documents() self.assertEqual(len(documents), len(document_ids)) for document in documents: self.assertIn(document.id, document_ids) for tag_id in selection: scope.remove_tag(tag_id) self.assertEqual(scope.get_concept_tags(), [])
def test_document_creation_with_single_tag(self): scope = Scope(database=self._database) scope.add_tag(2) self.assertEqual(scope.get_concept_tag_ids(), [2]) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 3, 4, 5, 6}) document = scope.create_document('main.py', 'python', '/tmp/main.py') self.assertEqual(set(scope.get_concept_document_ids()), {1, 4, 5, document.id}) self.assertEqual(scope.get_concept_tag_ids(), [2]) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 4, 5, document.id}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 3, 4, 5, 6}) scope.select_document(document.id) self.assertEqual(set(scope.get_concept_document_ids()), {1, 4, 5, document.id}) self.assertEqual(scope.get_concept_tag_ids(), [2]) self.assertEqual(scope.get_selection_document_ids(), [document.id]) self.assertEqual(scope.get_selection_tag_ids(), [2]) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 4, 5}) self.assertEqual(scope.get_selection_only_tag_ids(), []) scope.deselect_document(document.id) self.assertEqual(set(scope.get_concept_document_ids()), {1, 4, 5, document.id}) self.assertEqual(scope.get_concept_tag_ids(), [2]) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 4, 5, document.id}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 3, 4, 5, 6}) scope.remove_tag(2) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, document.id}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, document.id}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 2, 3, 4, 5, 6}) scope.select_document(document.id) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, document.id}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), [document.id]) self.assertEqual(scope.get_selection_tag_ids(), [2]) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(scope.get_selection_only_tag_ids(), [2])
def test_tag_removing_without_selection(self): scope = Scope(database=self._database) self.assertEqual(len(scope.get_concept_tags()), 0) for tag_id in range(1, 7): scope.add_tag(tag_id) self.assertEqual(len(scope.get_concept_tags()), 6) for tag_id in range(1, 7): scope.remove_tag(tag_id) self.assertEqual(len(scope.get_concept_tags()), 6 - tag_id)
def test_document_copy_to_root(self): scope = Scope(database=self._database) scope.copy_document(7) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 2, 3, 4, 5, 6}) scope.select_document(7) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), [7]) self.assertEqual(set(scope.get_selection_tag_ids()), {3, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 8}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {3, 6}) scope.add_tag(3) self.assertEqual(set(scope.get_concept_document_ids()), {2, 7, 8}) self.assertEqual(scope.get_concept_tag_ids(), [3]) self.assertEqual(scope.get_selection_document_ids(), [7]) self.assertEqual(set(scope.get_selection_tag_ids()), {3, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {2, 8}) self.assertEqual(scope.get_selection_only_tag_ids(), [6]) scope.remove_tag(3) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), [7]) self.assertEqual(set(scope.get_selection_tag_ids()), {3, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 8}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {3, 6}) scope.add_tag(6) self.assertEqual(set(scope.get_concept_document_ids()), {4, 7}) self.assertEqual(scope.get_concept_tag_ids(), [6]) self.assertEqual(scope.get_selection_document_ids(), [7]) self.assertEqual(set(scope.get_selection_tag_ids()), {3, 6}) self.assertEqual(scope.get_concept_only_document_ids(), [4]) self.assertEqual(scope.get_selection_only_tag_ids(), [3]) scope.remove_tag(6) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), [7]) self.assertEqual(set(scope.get_selection_tag_ids()), {3, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 8}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {3, 6})
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_destroying(self): scope = Scope(database=self._database) document_counts = { 1: [4, 3, 2, 1, 1, 0, 0, 0, 0], 2: [3, 2, 2, 2, 1, 0, 0, 0, 0], 3: [3, 3, 2, 2, 2, 2, 2, 1, 0], 4: [2, 2, 2, 1, 1, 1, 0, 0, 0], 5: [3, 3, 3, 3, 3, 2, 1, 1, 0], 6: [2, 2, 2, 2, 1, 1, 1, 0, 0] } for i in range(9): if i > 0: scope.destroy_document(i) for tag_id in range(1, 7): scope.add_tag(tag_id) self.assertEqual(len(scope.get_concept_documents()), document_counts[tag_id][i]) scope.remove_tag(tag_id)
def test_get_documents_with_single_tag(self): scope = Scope(database=self._database) document_ids = { 1: [1, 2, 3, 5], 2: [1, 4, 5], 3: [2, 7, 8], 4: [3, 6], 5: [5, 6, 8], 6: [4, 7] } for tag_id, document_ids in document_ids.items(): self.assertEqual(scope.get_concept_tags(), []) scope.add_tag(tag_id) documents = scope.get_concept_documents() self.assertEqual(len(documents), len(document_ids)) for document in documents: self.assertIn(document.id, document_ids) scope.remove_tag(tag_id) self.assertEqual(scope.get_concept_tags(), [])
def test_document_clone_with_single_tag(self): scope = Scope(database=self._database) scope.add_tag(6) self.assertEqual(set(scope.get_concept_document_ids()), {4, 7}) self.assertEqual(scope.get_concept_tag_ids(), [6]) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 2, 3, 4, 5}) cloned_document = scope.clone_document(2) self.assertEqual(set(scope.get_concept_document_ids()), {4, 7, cloned_document.id}) self.assertEqual(scope.get_concept_tag_ids(), [6]) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {4, 7, cloned_document.id}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 2, 3, 4, 5}) scope.remove_tag(6) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, cloned_document.id}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, cloned_document.id}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 2, 3, 4, 5, 6}) scope.select_document(cloned_document.id) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, cloned_document.id}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), [cloned_document.id]) self.assertEqual(scope.get_selection_tag_ids(), [6]) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(scope.get_selection_only_tag_ids(), [6])
def test_invalid_tag_removing(self): scope = Scope(database=self._database) with self.assertRaises(ValueError): scope.remove_tag(11) with self.assertRaises(ValueError): scope.remove_tag(0)
def test_document_clone_with_multiple_tags(self): scope = Scope(database=self._database) scope.add_tag(2) scope.add_tag(3) scope.add_tag(6) self.assertEqual(scope.get_concept_document_ids(), []) self.assertEqual(set(scope.get_concept_tag_ids()), {2, 3, 6}) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(scope.get_concept_only_document_ids(), []) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 4, 5}) cloned_document = scope.clone_document(6) self.assertNotEqual(cloned_document.id, 6) self.assertEqual(scope.get_concept_document_ids(), [cloned_document.id]) self.assertEqual(set(scope.get_concept_tag_ids()), {2, 3, 6}) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(scope.get_concept_only_document_ids(), [cloned_document.id]) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 4, 5}) scope.select_document(cloned_document.id) self.assertEqual(scope.get_concept_document_ids(), [cloned_document.id]) self.assertEqual(set(scope.get_concept_tag_ids()), {2, 3, 6}) self.assertEqual(scope.get_selection_document_ids(), [cloned_document.id]) self.assertEqual(set(scope.get_selection_tag_ids()), {2, 3, 6}) self.assertEqual(scope.get_concept_only_document_ids(), []) self.assertEqual(scope.get_selection_only_tag_ids(), []) scope.remove_tag(3) self.assertEqual(set(scope.get_concept_document_ids()), {4, cloned_document.id}) self.assertEqual(set(scope.get_concept_tag_ids()), {2, 6}) self.assertEqual(scope.get_selection_document_ids(), [cloned_document.id]) self.assertEqual(set(scope.get_selection_tag_ids()), {2, 3, 6}) self.assertEqual(scope.get_concept_only_document_ids(), [4]) self.assertEqual(scope.get_selection_only_tag_ids(), [3]) scope.remove_tag(2) self.assertEqual(set(scope.get_concept_document_ids()), {4, 7, cloned_document.id}) self.assertEqual(set(scope.get_concept_tag_ids()), {6}) self.assertEqual(scope.get_selection_document_ids(), [cloned_document.id]) self.assertEqual(set(scope.get_selection_tag_ids()), {2, 3, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {4, 7}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {2, 3}) scope.remove_tag(6) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, cloned_document.id}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), [cloned_document.id]) self.assertEqual(set(scope.get_selection_tag_ids()), {2, 3, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {2, 3, 6}) scope.deselect_document(cloned_document.id) self.assertEqual(set(scope.get_concept_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, cloned_document.id}) self.assertEqual(scope.get_concept_tag_ids(), []) self.assertEqual(scope.get_selection_document_ids(), []) self.assertEqual(set(scope.get_selection_tag_ids()), {1, 2, 3, 4, 5, 6}) self.assertEqual(set(scope.get_concept_only_document_ids()), {1, 2, 3, 4, 5, 6, 7, 8, cloned_document.id}) self.assertEqual(set(scope.get_selection_only_tag_ids()), {1, 2, 3, 4, 5, 6}) document_counts = {1: 4, 2: 3 + 1, 3: 3 + 1, 4: 2, 5: 3, 6: 2 + 1} for tag_id, document_count in document_counts.items(): scope.add_tag(tag_id) self.assertEqual(len(scope.get_concept_documents()), document_count) scope.remove_tag(tag_id)