Skip to content

Commit c4b921f

Browse files
committed
Update test_timeseries_model.py
1 parent ea5b640 commit c4b921f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/test_pownet/test_stochastic/test_timeseries_model.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,14 @@
11
"""test_timeseries_model.py"""
22

33
import unittest
4+
5+
# Skip this test if the user has not installed pmdarima package
6+
try:
7+
import pmdarima
8+
except ImportError:
9+
pmdarima = None
10+
raise unittest.SkipTest("pmdarima is not installed, skipping tests.")
11+
412
import pandas as pd
513
from pownet.stochastic import timeseries_model
614

0 commit comments

Comments
 (0)