experimental: Ledger changed to Journal

core/test_uncore.py

@@ -1,56 +1,53 @@
 import pytest
-from uncore import ChartDict, Chart, Contra, Regular, T, Asset
-
-# test case: overwrite any reserved name in Chart
+from uncore import (
+    Account,
+    Chart,
+    ChartDict,
+    Contra,
+    Intermediate,
+    Journal,
+    Reference,
+    Regular,
+    Side,
+    T,
+    credit,
+    debit,
+    double_entry,
+)
 
 
 def test_contra_account_added_to_chart():
-    chart_dict = (
-        ChartDict().add_regular(T.Income, "sales").add_offset("sales", "refunds")
-    )
-    assert chart_dict["sales"] == Regular(T.Income)
-    assert chart_dict["refunds"] == Contra("sales")
-
-
-def test_invalid_contra_account_not_added_to_chart_by_assignment():
-    chart_dict = ChartDict()
-    with pytest.raises(KeyError):
-        chart_dict["voids"] = Contra("sales")
+    chart_dict = ChartDict().add(T.Income, "sales").offset("sales", "refunds")
+    assert chart_dict["sales"] == T.Income
+    assert chart_dict["refunds"] == Reference("sales")
 
 
 def test_invalid_contra_account_not_added_to_chart_by_method():
     with pytest.raises(KeyError):
-        ChartDict().add_offset("sales", "voids")
+        ChartDict().offset("sales", "voids")
 
 
 def test_chart_regular_names():
     chart_dict = (
-        ChartDict()
-        .add_regular(T.Asset, "cash")
-        .add_regular(T.Capital, "equity")
-        .add_offset("equity", "ts")
+        ChartDict().add(T.Asset, "cash").add(T.Capital, "equity").offset("equity", "ts")
     )
-    assert chart_dict.regular_names([T.Asset]) == ["cash"]
-    assert chart_dict.regular_names([T.Capital]) == ["equity"]
-    assert chart_dict.regular_names([T.Asset, T.Capital]) == ["cash", "equity"]
+    assert chart_dict.regular_names(T.Asset) == ["cash"]
+    assert chart_dict.regular_names(T.Capital) == ["equity"]
+    assert chart_dict.regular_names(T.Asset, T.Capital) == ["cash", "equity"]
 
 
 def test_chart_dict_contra_pairs():
-    chart_dict = ChartDict().add_regular(T.Capital, "equity").add_offset("equity", "ts")
+    chart_dict = ChartDict().add(T.Capital, "equity").offset("equity", "ts")
     assert chart_dict.contra_pairs(T.Capital) == [("equity", "ts")]
 
 
 def test_chart_contra_pairs():
-    chart = (
-        Chart()
-        .add(T.Capital, "equity", contra_names=["buyback"])
-        .add(T.Income, "sales", contra_names=["refunds", "voids"])
-    )
-    assert chart.dict.contra_pairs(T.Income) == [
+    assert Chart().add(
+        T.Income, "sales", contra_names=["refunds", "voids"]
+    ).dict.contra_pairs(T.Income) == [
         ("sales", "refunds"),
         ("sales", "voids"),
     ]
-    assert chart.dict.contra_pairs(T.Capital) == [("equity", "buyback")]
 
 
 def test_chart_creation():
@@ -60,20 +57,95 @@ def test_chart_creation():
         .add(T.Asset, "cash")
         .add(T.Capital, "equity", contra_names=["buyback"])
     )
-    chart_dict0 = ChartDict(
+    assert chart0.dict ==  ChartDict(
         {
-            "cash": Regular(t=T.Asset),
-            "sales": Regular(t=T.Income),
-            "refunds": Contra(links_to="sales"),
-            "voids": Contra(links_to="sales"),
-            "equity": Regular(t=T.Capital),
-            "buyback": Contra(links_to="equity"),
+            "cash": T.Asset,
+            "sales": T.Income,
+            "refunds": Reference("sales"),
+            "voids": Reference("sales"),
+            "equity": T.Capital,
+            "buyback": Reference("equity"),
         }
     )
-    assert chart0.dict == chart_dict0
     assert chart0.retained_earnings_account == "retained_earnings"
     assert chart0.income_summary_account == "income_summary_account"
 
 
-def test_condense():
-    assert Asset([10, 20], [5]).condense() == Asset([25], [])
+def test_journal_creation():
+    assert Journal.from_chart(
+        ChartDict(cash=T.Asset, contra_cash=Reference("cash")), "isa", "re"
+    ) == Journal(
+        {
+            "cash": Account(Regular(T.Asset)),
+            "contra_cash": Account(Contra(T.Asset)),
+            "re": Account(Regular(T.Capital)),
+            "isa": Account(Intermediate(Side.Credit)),
+        }
+    )
+
+
+@pytest.fixture
+def journal():
+    chart = (
+        Chart()
+        .add_many(T.Asset, "cash", "ar")
+        .add(T.Capital, "equity", contra_names=["buyback"])
+        .add(T.Income, "sales", contra_names=["refunds", "voids"])
+        .add(T.Liability, "vat")
+        .add(T.Expense, "salary")
+    )
+    ledger = Journal.from_chart(
+        chart.dict, "income_summary_account", "retained_earnings"
+    )
+    ledger.post_many(
+        [
+            double_entry("cash", "equity", 1200),
+            double_entry("buyback", "cash", 200),
+            [debit("ar", 70), credit("sales", 60), credit("vat", 10)],
+            double_entry("cash", "ar", 30),
+            [
+                debit("voids", 20),
+                debit("refunds", 10),
+                debit("vat", 5),
+                credit("ar", 35),
+            ],
+            [debit("salary", 18), credit("cash", 18)],
+        ]
+    )
+    return ledger
+
+
+def test_balances(journal):
+    assert journal.balances == {
+        "cash": 1012,
+        "ar": 5,
+        "equity": 1200,
+        "buyback": 200,
+        "sales": 60,
+        "refunds": 10,
+        "voids": 20,
+        "vat": 5,
+        "salary": 18,
+        "retained_earnings": 0,
+        "income_summary_account": 0,
+    }
+
+
+def test_tuples(journal):
+    assert journal.tuples == {
+        "cash": (1230, 218),
+        "ar": (70, 65),
+        "equity": (0, 1200),
+        "buyback": (200, 0),
+        "sales": (0, 60),
+        "refunds": (10, 0),
+        "voids": (20, 0),
+        "vat": (5, 10),
+        "salary": (18, 0),
+        "retained_earnings": (0, 0),
+        "income_summary_account": (0, 0),
+    }
+
+
+def test_subset(journal):
+    assert journal.subset(T.Asset).balances == {"cash": 1012, "ar": 5}