Bugfix

lkilcher · Dec 7, 2022 · 645c1a6 · 645c1a6
1 parent a716476
commit 645c1a6
Show file tree

Hide file tree

Showing 9 changed files with 11 additions and 11 deletions.
diff --git a/dolfyn/tests/data/RDI_7f79.nc b/dolfyn/tests/data/RDI_7f79.nc
diff --git a/dolfyn/tests/data/RDI_test01.nc b/dolfyn/tests/data/RDI_test01.nc
diff --git a/dolfyn/tests/data/RDI_withBT.nc b/dolfyn/tests/data/RDI_withBT.nc
diff --git a/dolfyn/tests/data/RiverPro_test01.nc b/dolfyn/tests/data/RiverPro_test01.nc
diff --git a/dolfyn/tests/data/vmdas01.nc b/dolfyn/tests/data/vmdas01.nc
diff --git a/dolfyn/tests/data/winriver01.nc b/dolfyn/tests/data/winriver01.nc
diff --git a/dolfyn/tests/data/winriver02.nc b/dolfyn/tests/data/winriver02.nc
diff --git a/dolfyn/tests/data/winriver02_transect.nc b/dolfyn/tests/data/winriver02_transect.nc
diff --git a/dolfyn/velocity.py b/dolfyn/velocity.py
@@ -916,7 +916,7 @@ def calc_tke(self, veldat, noise=[0, 0, 0], detrend=True):
         else:
             vel = self.demean(vel)
 
-        if 'b5' in veldat.name:
+        if 'time_b5' in veldat.dims:
             time = self.mean(veldat.time_b5.values)
         else:
             time = self.mean(veldat.time.values)
@@ -938,7 +938,7 @@ def calc_tke(self, veldat, noise=[0, 0, 0], detrend=True):
             da = da.assign_coords({'tke': ["upup_", "vpvp_", "wpwp_"],
                                    'time': time})
         else:
-            if 'b5' in veldat.name:
+            if 'time_b5' in veldat.dims:
                 da = da.assign_coords({'time_b5': time})
             else:
                 da = da.assign_coords({'time': time})
@@ -986,7 +986,7 @@ def calc_psd(self, veldat,
           The spectra in the 'u', 'v', and 'w' directions.
         """
 
-        if 'b5' in veldat.name:
+        if 'time_b5' in veldat.dims:
             time = self.mean(veldat.time_b5.values)
             time_str = 'time_b5'
         else: