fixed test bug

Transport-for-the-North · Jan 8, 2025 · 55b54e2 · 55b54e2
1 parent 2d34008
commit 55b54e2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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)