From a555f34c477e1f2eb4157999a629c583fbb90ca7 Mon Sep 17 00:00:00 2001
From: Florent Fourcot <weboob@flo.fourcot.fr>
Date: Thu, 29 Dec 2016 23:15:39 +0100
Subject: [PATCH] iban-tools: replace print in __main__ with a real test suite

---
 setup.cfg                              | 3 ++-
 weboob/tools/capabilities/bank/iban.py | 9 +++++----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index 86da6eb73d..fbbe7c4d6a 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -4,7 +4,8 @@ detailed-errors = 1
 with-doctest = 1
 with-coverage = 1
 where = weboob
-tests = weboob.tools.capabilities.bank.transactions,
+tests = weboob.tools.capabilities.bank.iban,
+        weboob.tools.capabilities.bank.transactions,
         weboob.tools.capabilities.paste,
         weboob.tools.application.formatters.json,
         weboob.tools.application.formatters.table,
diff --git a/weboob/tools/capabilities/bank/iban.py b/weboob/tools/capabilities/bank/iban.py
index 1f63d59e9c..3292ab377f 100644
--- a/weboob/tools/capabilities/bank/iban.py
+++ b/weboob/tools/capabilities/bank/iban.py
@@ -91,7 +91,8 @@ def rebuild_rib(rib):
     key = find_rib_checksum(rib[:5], rib[5:10], rib[10:21])
     return unicode(rib[:21] + ('%02d' % key))
 
-if __name__ == '__main__':
-    print(rebuild_iban('FR0013048379405300290000355'))
-    print(rebuild_iban('GB87BARC20658244971655'))
-    print(rebuild_rib('30003021990005077567600'))
+
+def test():
+    assert rebuild_iban('FR0013048379405300290000355') == "FR7613048379405300290000355"
+    assert rebuild_iban('GB87BARC20658244971655') == "GB87BARC20658244971655"
+    assert rebuild_rib('30003021990005077567600') == "30003021990005077567667"
-- 
GitLab