Changeset 9372

Timestamp:

21 Oct 2012, 10:43:54 (12 years ago)

Author:

uli

Message:

Add a helper to provide a (Python 3 compatible) product calculator for sequences of numbers.

Location:

main/waeup.kofa/trunk/src/waeup/kofa/utils

Files:

• helpers.py (modified) (1 diff)
• tests/test_helpers.py (modified) (3 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/utils/helpers.py

@@ -685 +685 @@
         writer.writerow(row)
     return tmp_path
+
+def product(sequence, start=1):
+    """Returns the product of a sequence of numbers (_not_ strings)
+    multiplied by the parameter `start` (defaults to 1). If the
+    sequence is empty, returns 0.
+    """
+    if not len(sequence):
+        return 0
+    result = start
+    for item in sequence:
+        result *= item
+    return result

main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_helpers.py

@@ -39 +39 @@
     implements(IFakeObject)
 
+class SimpleHelpersTestCase(unittest.TestCase):
+    # Tests for simple functions in `helpers`.
+    def test_product(self):
+        # the product will return zero without input
+        result1 = helpers.product([])
+        result2 = helpers.product([1,2,3])
+        result3 = helpers.product([], start=5)
+        result4 = helpers.product([1,2,3], start=5)
+        self.assertEqual(result1, 0)
+        self.assertEqual(result2, 6)
+        self.assertEqual(result3, 0)
+        self.assertEqual(result4, 30)
+        return
+
 class RemoveFileOrDirectoryTestCase(unittest.TestCase):
 
@@ -447 +461 @@
             contents, '\r\n')
         return
+
 
 def test_suite():
@@ -463 +478 @@
         GetFileFormatTestCase,
         MergeCSVFileTestCase,
+        SimpleHelpersTestCase,
         ]:
         suite.addTests(