def testTruncate(self): tests = [(('1234567', 5), '12...'), (('1234567', 5, True), '12...'), (('1234567', 5, False), '...67')] for test, res in tests: self.assertEqual(truncate(*test), res)
def testNoTruncate(self): tests = [('0', 4, True), ('1234', 4, False)] for test in tests: self.assertIs(truncate(*test), test[0])
def testError(self): tests = [('', 3, True), ('0', -1, False)] for test in tests: with self.assertRaises(ValueError): truncate(*test)
def test_truncate(test, res): assert truncate(*test) == res
def test_truncate_noop(test): assert truncate(*test) is test[0]
def test_truncate_error(test): with pytest.raises(ValueError): truncate(*test)