def test_format_claim_special(self): claim = pywikibot.Claim(self.repo, 'P123') claim.setSnakType('novalue') expected = 'wd_template_1: formatted_itis' self.assertEqual(PreviewItem.format_claim(claim), expected) self.mock_wd_template.assert_called_once_with('P123') self.mock_format_itis.assert_called_once_with('novalue', True)
def test_format_claim_basic(self): claim = pywikibot.Claim(self.repo, 'P123') claim.setTarget('1') expected = 'wd_template_1: formatted_itis' self.assertEqual(PreviewItem.format_claim(claim), expected) self.mock_wd_template.assert_called_once_with('P123') self.mock_format_itis.assert_called_once_with('1', False)