From 55b54e2e151732e149624f743f344ce9ae55f7c4 Mon Sep 17 00:00:00 2001
From: Kieran-Fishwick-TfN <kieran.fishwick@transportforthenorth.com>
Date: Wed, 8 Jan 2025 12:08:30 +0000
Subject: [PATCH] fixed test bug

---
 tests/gravity_model/test_multi_area.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tests/gravity_model/test_multi_area.py b/tests/gravity_model/test_multi_area.py
index 274f7da..adfe256 100644
--- a/tests/gravity_model/test_multi_area.py
+++ b/tests/gravity_model/test_multi_area.py
@@ -4,11 +4,10 @@
 from pathlib import Path
 
 # Third Party
-from matplotlib import pyplot as plt
 import numpy as np
 import pandas as pd
 import pytest
-from caf.toolkit import cost_utils
+from matplotlib import pyplot as plt
 
 # Local Imports
 from caf.distribute import cost_functions
@@ -249,5 +248,5 @@ def test_results(self, cal_results, request):
         assert isinstance(cal_results, dict)
         for result in cal_results.values():
             assert isinstance(result, GravityModelResults)
-            assert isinstance(result.summary(), pd.Series)
+            assert isinstance(result.summary, pd.Series)
             assert isinstance(result.plot_distributions(), plt.Figure)