Merge pull request #40 from jamilatta/tk6

Tk6 - refurbished
scieloorg · Jun 23, 2017 · 04129c7 · 04129c7
2 parents 0bea84d + 31259c9
commit 04129c7
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/legendarium/legendarium.py b/legendarium/legendarium.py
@@ -73,7 +73,10 @@ def _clean_suppl_number(self):
         """
         Clean the suppl_number stripped the beginning and the end of the string.
         """
-        return self.suppl_number.strip()
+        if self.suppl_number:
+            return self.suppl_number.strip()
+        else:
+            return ''
 
     def get_journal(self):
         """

diff --git a/legendarium/urlegendarium.py b/legendarium/urlegendarium.py
@@ -73,7 +73,10 @@ def _clean_suppl_number(self):
         """
         Clean the supplement number stripped the beginning and the end of the string.
         """
-        return self.suppl_number.strip()
+        if self.suppl_number:
+            return self.suppl_number.strip()
+        else:
+            return ''
 
     def get_journal_seg(self):
         """

diff --git a/tests/test_legendarium.py b/tests/test_legendarium.py
@@ -78,7 +78,7 @@ def test_build_legend_without_volume_and_number_and_year_pub_and_suppl_number(se
 
         self.assertEqual(u'Rev.Mal-Estar Subj:e00120416', leg.stamp)
 
-    def test_build_legend_without_volume_and_number_and_year_pub_(self):
+    def test_build_legend_without_volume_and_number_and_year_pub(self):
 
         del(self.dict_leg['volume'])  # Remove the volume
         del(self.dict_leg['number'])  # Remove the number