Add custom type

1 files changed, 3 insertions, 2 deletions

diff --git a/ledger_process.py b/ledger_process.py
index dd82877..adcaacb 100644
--- a/ledger_process.py
+++ b/ledger_process.py
@@ -3,6 +3,7 @@ from itertools import groupby
 import logging
 from typing import Dict, List
 
+from exceptions import MismatchedTradeError
 from ledger_action import LedgerAction
 from trade import Trade
 from trade_queue import FIFOQueue
@@ -98,7 +99,7 @@ class LedgerProcess:
                     )
             else:
                 logger.error(f"Trade group doesn't have expected currencies.")
-                raise ValueError(f"Unexpected trade grouping for refid {refid}")
+                raise MismatchedTradeError(f"Unexpected trade grouping for refid {refid}")
         else:
             logger.error(f"Trade group has {len(trades)} trades, expected 2.")
-            raise ValueError(f"Unexpected number of trades for refid {refid}")
+            raise MismatchedTradeError(f"Unexpected number of trades for refid {refid}")