From 5b99a92f05794db6ae172e35f9733131c0229072 Mon Sep 17 00:00:00 2001 From: Marlon Rodriguez Garcia Date: Tue, 19 Nov 2024 14:36:05 -0500 Subject: [PATCH] Add test to improve coverage --- num2words/base.py | 9 +++------ tests/test_base.py | 13 +++++++++++++ 2 files changed, 16 insertions(+), 6 deletions(-) diff --git a/num2words/base.py b/num2words/base.py index 243045ff..26f78edd 100644 --- a/num2words/base.py +++ b/num2words/base.py @@ -39,15 +39,12 @@ def __init__(self): self.errmsg_floatord = "Cannot treat float %s as ordinal." self.errmsg_negord = "Cannot treat negative num %s as ordinal." self.errmsg_toobig = "abs(%s) must be less than %s." + self.cards = OrderedDict() + self.set_numwords() + self.MAXVAL = 1000 * list(self.cards.keys())[0] self.setup() - # uses cards - if any(hasattr(self, field) for field in - ['high_numwords', 'mid_numwords', 'low_numwords']): - self.cards = OrderedDict() - self.set_numwords() - self.MAXVAL = 1000 * list(self.cards.keys())[0] def set_numwords(self): self.set_high_numwords(self.high_numwords) diff --git a/tests/test_base.py b/tests/test_base.py index efa1bc81..e05bc7df 100644 --- a/tests/test_base.py +++ b/tests/test_base.py @@ -62,3 +62,16 @@ def test_is_title(self): self.base.title("one"), "one" ) + + def test_set_high_numwords_not_implemented(self): + with self.assertRaises(NotImplementedError): + self.base.set_high_numwords() + + def test_to_ordinal(self): + from num2words.base import Num2Word_Base + self.base = Num2Word_Base() + self.assertEqual(self.base.to_ordinal(1), "1st") + self.assertEqual(self.base.to_ordinal(11), "11th") + self.assertEqual(self.base.to_ordinal(21), "21st") + self.assertEqual(self.base.to_ordinal(100), "100th") + self.assertEqual(self.base.to_ordinal(123), "123rd")