def test_rank(self):
self.assertEqual(self.claim1.getRank(), 'normal')
claim = Claim('P134', datatype='wikibase-item')
claim.setRank('preferred')
self.assertEqual(claim.getRank(), 'preferred')
self.assertEqual(claim.rank, 'preferred')
def test_set_target(self):
claim = Claim('P144', datatype='wikibase-item')
self.assertRaises(ValueError, claim.setTarget, "Hello!")
claim = Claim('P144', datatype='time')
self.assertRaises(ValueError, claim.setTarget, ItemPage('Q5'))
self.assertRaises(ValueError, claim.setTarget, "Hello!")
def test_add_claim(self):
claim = Claim('P17', datatype='wikibase-item')
claim.setTarget(ItemPage('Q91'))
self.item_page.addClaim(claim)
self.assertIn('P17', self.item_page.claims)
self.assertEqual(len(self.item_page.claims['P17']), 1)
self.assertIsInstance(self.item_page.claims['P17'][0], Claim)
def test_snak_type(self):
self.assertEqual(self.claim1.snaktype, 'value')
self.assertRaises(ValueError, self.claim1.setSnakType, 'NULL')
self.assertRaises(ValueError, self.claim1.setSnakType, None)
claim = Claim('P31')
claim.setSnakType('novalue')
self.assertEqual(claim.snaktype, 'novalue')
self.assertNotEqual(claim.snaktype, 'value')
def test_qualifiers(self):
source = self.claim1.sources[0]['P143'][0]
self.assertRaises(ValueError, source.has_qualifier, 'P31', 'Q5')
qualifier = Claim('P31', datatype='wikibase-item')
qualifier.setTarget(ItemPage('Q5'))
self.claim1.addQualifier(qualifier)
self.assertRaises(ValueError, source.addQualifier, qualifier)
self.assertEqual(self.claim1.qualifiers['P31'][0], qualifier)
self.assertTrue(qualifier.isQualifier)
self.assertTrue(self.claim1.has_qualifier('P31', 'Q5'))
self.assertFalse(self.claim1.has_qualifier('P31', 'Q6'))
self.assertFalse(self.claim1.has_qualifier('P32', 'Q5'))
def test_sources(self):
self.assertIsInstance(self.claim1.sources, list)
self.assertIsInstance(self.claim1.sources[0], OrderedDict)
self.assertIn('P143', self.claim1.sources[0])
self.assertIsInstance(self.claim1.sources[0]['P143'], list)
self.assertEqual(len(self.claim1.sources), 2)
self.assertEqual(self.claim1.getSources(), self.claim1.sources)
sources_copy = self.claim1.sources
source = Claim('P144', datatype='wikibase-item')
source.setTarget(ItemPage('Q5'))
self.claim1.addSource(source)
self.assertIn('P144', self.claim1.sources[-1])
self.assertEqual(source, self.claim1.sources[-1]['P144'][0])
self.claim1.removeSource(source)
self.assertEqual(self.claim1.sources, sources_copy)
