def test_inplace_and(self): s1 = EntitySet((self.t2, self.t1)) s1 &= EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s1.ids), [self.t2.id]) self.assertIsNot(s1._ontology, None) s2 = EntitySet((self.t2, self.t1)) s2 &= {self.t2, self.t3} self.assertEqual(sorted(s2.ids), [self.t2.id]) self.assertIsNot(s2._ontology, None) s3 = EntitySet() s3 &= {} self.assertEqual(sorted(s3.ids), []) self.assertIs(s3._ontology, None)
def test_and(self): s1 = EntitySet((self.t2, self.t1)) s = s1 & EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s.ids), [self.t2.id]) self.assertIsNot(s._ontology, None) s2 = EntitySet((self.t2, self.t1)) s = s2 & {self.t2, self.t3} self.assertEqual(sorted(s.ids), [self.t2.id]) self.assertIsNot(s._ontology, None) s3 = EntitySet() s = s3 & {} self.assertEqual(sorted(s.ids), []) self.assertIs(s._ontology, None)
def test_contains(self): s = EntitySet({self.t1, self.t2}) self.assertIn(self.t1, s) self.assertIn(self.t2, s) self.assertNotIn(self.t3, s) self.assertIn(self.ont[self.t1.id], s) self.assertIn(self.ont[self.t2.id], s) self.assertNotIn(self.ont[self.t3.id], s)
def test_add_empty(self): s1 = EntitySet() self.assertEqual(len(s1), 0) s1.add(self.t1) s1.add(self.t2) self.assertEqual(len(s1), 2) self.assertIn(s1.pop(), {self.t1, self.t2}) self.assertIn(s1.pop(), {self.t1, self.t2})
def test_sub(self): s1 = EntitySet((self.t2, self.t1)) s = s1 - EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s.ids), [self.t1.id]) self.assertIsNot(s._ontology, None) s2 = EntitySet((self.t2, self.t1)) s = s2 - {self.t2, self.t3} self.assertEqual(sorted(s.ids), [self.t1.id]) self.assertIsNot(s._ontology, None) s3 = EntitySet() s = s3 - EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s.ids), []) self.assertIs(s._ontology, None) s4 = EntitySet() s = s4 - {self.t2, self.t3} self.assertEqual(sorted(s.ids), []) self.assertIs(s._ontology, None)
def test_or(self): s1 = EntitySet((self.t2, self.t1)) s = s1 | EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s.ids), [self.t1.id, self.t2.id, self.t3.id]) self.assertIsNot(s._ontology, None) s2 = EntitySet((self.t2, self.t1)) s = s2 | {self.t2, self.t3} self.assertEqual(sorted(s.ids), [self.t1.id, self.t2.id, self.t3.id]) self.assertIsNot(s._ontology, None) s3 = EntitySet() s = s3 | EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s.ids), [self.t2.id, self.t3.id]) self.assertIsNot(s._ontology, None) s4 = EntitySet() s = s4 | {self.t2, self.t3} self.assertEqual(sorted(s.ids), [self.t2.id, self.t3.id]) self.assertIsNot(s._ontology, None)
def test_inplace_sub(self): s1 = EntitySet((self.t2, self.t1)) s1 -= EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s1.ids), [self.t1.id]) self.assertIsNot(s1._ontology, None) s2 = EntitySet((self.t2, self.t1)) s2 -= {self.t2, self.t3} self.assertEqual(sorted(s2.ids), [self.t1.id]) self.assertIsNot(s2._ontology, None) s3 = EntitySet() s3 -= EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s3.ids), []) self.assertIs(s3._ontology, None) s4 = EntitySet() s4 -= {self.t2, self.t3} self.assertEqual(sorted(s4.ids), []) self.assertIs(s4._ontology, None)
def test_inplace_xor(self): s1 = EntitySet((self.t2, self.t1)) s = s1 ^ EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s.ids), [self.t1.id, self.t3.id]) self.assertIsNot(s._ontology, None) s2 = EntitySet((self.t2, self.t1)) s = s2 ^ {self.t2, self.t3} self.assertEqual(sorted(s.ids), [self.t1.id, self.t3.id]) self.assertIsNot(s._ontology, None) s3 = EntitySet() s = s3 ^ EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s.ids), [self.t2.id, self.t3.id]) self.assertIsNot(s._ontology, None) s4 = EntitySet({self.t2, self.t3}) s = s4 ^ s4 self.assertEqual(sorted(s.ids), []) self.assertIs(s._ontology, None)
def test_inplace_or(self): s1 = EntitySet((self.t2, self.t1)) s1 |= EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s1.ids), [self.t1.id, self.t2.id, self.t3.id]) self.assertIsNot(s1._ontology, None) s2 = EntitySet((self.t2, self.t1)) s2 |= {self.t2, self.t3} self.assertEqual(sorted(s2.ids), [self.t1.id, self.t2.id, self.t3.id]) self.assertIsNot(s2._ontology, None) s3 = EntitySet() s3 |= EntitySet((self.t2, self.t3)) self.assertEqual(sorted(s3.ids), [self.t2.id, self.t3.id]) self.assertIsNot(s3._ontology, None) s4 = EntitySet() s4 |= {self.t2, self.t3} self.assertEqual(sorted(s4.ids), [self.t2.id, self.t3.id]) self.assertIsNot(s4._ontology, None)