From 348c435d1ed10acf308bbbd53cdca9bc1213282b Mon Sep 17 00:00:00 2001 From: Jim Paris Date: Mon, 1 Apr 2013 16:38:50 -0400 Subject: [PATCH] Catch arctan2 errors --- src/sinefit.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/sinefit.py b/src/sinefit.py index 19e0c2a..88ce0fe 100755 --- a/src/sinefit.py +++ b/src/sinefit.py @@ -172,7 +172,11 @@ def sfit4(data, fs): ## Extract results A = sqrt(s[0]*s[0] + s[1]*s[1]) # eqn B.21 f0 = w / (2*pi) - phi = -arctan2(s[1], s[0]) # eqn B.22 + try: + phi = -arctan2(s[1], s[0]) # eqn B.22 + except TypeError: + # something broke down, just return zeros + return (0, 0, 0, 0) C = s[2] return (A, f0, phi, C)