diff --git a/HARK/tests/test_HARKutilities.py b/HARK/tests/test_HARKutilities.py index 95be07999..1e945d513 100644 --- a/HARK/tests/test_HARKutilities.py +++ b/HARK/tests/test_HARKutilities.py @@ -3,9 +3,9 @@ """ # Bring in modules we need import unittest -import matplotlib.pyplot as plt from types import SimpleNamespace +import matplotlib.pyplot as plt import numpy as np from HARK.rewards import ( @@ -84,10 +84,10 @@ def test_asset_grid(self): test = np.unique(np.diff(aXtraGrid).round(decimals=3)) self.assertEqual(test.size, 1) - + def test_make_figs(self): # Test the make_figs() function with a trivial output plt.figure() - plt.plot(np.linspace(1,5,40),np.linspace(4,8,40) ) - make_figs('test', True , False, target_dir='') + plt.plot(np.linspace(1, 5, 40), np.linspace(4, 8, 40)) + make_figs("test", True , False, target_dir="") plt.clf()