def test_round_record(self):
normalizer = RoundNormalizer(attr_index=1)
self.assertEqual(['a1', '3'], normalizer.normalize(['a1', 3.14159]))
normalizer = RoundNormalizer(attr_index=1, ndigits=2)
self.assertEqual(['a1', '3.14'], normalizer.normalize(['a1', 3.14159]))
normalizer = RoundNormalizer(attr_index=1, ndigits=3)
self.assertEqual(['a1', '3.142'], normalizer.normalize(['a1', 3.14159]))
def test_round(self):
normalizer = RoundNormalizer()
self.assertEqual(normalizer.normalize(3.14159), '3')
normalizer = RoundNormalizer(ndigits=2)
self.assertEqual(normalizer.normalize(3.14159), '3.14')
normalizer = RoundNormalizer(ndigits=3)
self.assertEqual(normalizer.normalize(3.14159), '3.142')