Added warning for time

giuliovv · Oct 30, 2021 · 473f7e4 · 473f7e4
1 parent 0823f24
commit 473f7e4
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/dist/tfest-0.1.11.tar.gz b/dist/tfest-0.1.11.tar.gz
diff --git a/tfest.egg-info/PKG-INFO b/tfest.egg-info/PKG-INFO
@@ -1,12 +1,12 @@
 Metadata-Version: 2.1
 Name: tfest
-Version: 0.1.10
+Version: 0.1.11
 Summary: Transfer function estimation based on frequency response.
 Home-page: https://github.com/giuliovv/tfest
 Author: Giulio Vaccari
 Author-email: io@giuliovaccari.it
 License: MIT
-Download-URL: https://github.com/giuliovv/tfest/archive/refs/tags/v0.1.10-alpha.tar.gz
+Download-URL: https://github.com/giuliovv/tfest/archive/refs/tags/v0.1.11-alpha.tar.gz
 Description: # Tfest
         Transfer function estimation with frequency response. Only works for SISO systems for now.
         Python equivalent of MATLAB tfest

diff --git a/tfest/tfest.py b/tfest/tfest.py
@@ -1,5 +1,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
+import warnings
 
 from matplotlib.mlab import psd, csd
 from scipy.optimize import minimize
@@ -33,13 +34,16 @@ def loss(self, x, nzeros, freq, H):
         risp /= np.array([np.polyval(poles, s) for s in 1j*freq])
         return np.linalg.norm((risp-H).reshape(-1, 1), axis=1).sum() #+ np.abs(x).sum()
 
-    def frequency_response(self, method="h1", time=1):
+    def frequency_response(self, method="h1", time=None):
         """
         method: "fft" or "h1" (default) or "h2"
         time: time for fft
 
         return: frequency response and frequency
         """
+        if time == None:
+            warnings.warn("Setting default time=1")
+            time = 1
         if method == "fft":
             u_f = fft(self.u)
             y_f = fft(self.y)