first commit

modules/account_asset/tests/test_module.py

@@ -0,0 +1,50 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+import datetime as dt
+
+from trytond.modules.account_asset.asset import normalized_delta
+from trytond.tests.test_tryton import ModuleTestCase
+
+
+class AccountAssetTestCase(ModuleTestCase):
+    'Test AccountAsset module'
+    module = 'account_asset'
+    extras = ['purchase']
+
+    def test_normalized_delta(self):
+        "Test normalized delta"
+        for start, end, delta in [
+                (dt.date(2019, 1, 1), dt.date(2019, 12, 31),
+                    dt.timedelta(days=364)),
+                (dt.date(2019, 1, 1), dt.date(2020, 1, 1),
+                    dt.timedelta(days=365)),
+                (dt.date(2019, 1, 1), dt.date(2019, 3, 1),
+                    dt.timedelta(days=31 + 28)),
+                (dt.date(2019, 2, 28), dt.date(2019, 3, 31),
+                    dt.timedelta(days=31)),
+                (dt.date(2024, 1, 1), dt.date(2024, 2, 1),
+                    dt.timedelta(days=31)),
+                (dt.date(2024, 1, 1), dt.date(2024, 3, 1),
+                    dt.timedelta(days=31 + 28)),
+                (dt.date(2024, 2, 27), dt.date(2024, 3, 31),
+                    dt.timedelta(days=32)),
+                (dt.date(2024, 2, 28), dt.date(2024, 3, 31),
+                    dt.timedelta(days=31)),
+                (dt.date(2024, 2, 29), dt.date(2024, 3, 31),
+                    dt.timedelta(days=31)),
+                (dt.date(2024, 3, 1), dt.date(2024, 4, 1),
+                    dt.timedelta(days=31)),
+                (dt.date(2024, 1, 1), dt.date(2025, 1, 1),
+                    dt.timedelta(days=365)),
+                (dt.date(2023, 1, 1), dt.date(2025, 1, 1),
+                    dt.timedelta(days=365 * 2)),
+                (dt.date(2000, 1, 1), dt.date(2020, 1, 1),
+                    dt.timedelta(days=365 * 20)),
+                ]:
+            with self.subTest(start=start, end=end):
+                self.assertEqual(
+                    normalized_delta(start, end), delta)
+
+
+del ModuleTestCase