Fix units

sngmanish · Nov 15, 2019 · 895c0f9 · 895c0f9
1 parent 8cad302
commit 895c0f9
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/heliopy/data/helper.py b/heliopy/data/helper.py
@@ -159,6 +159,8 @@ def cdf_dict(unit_string):
                         ('None', u.dimensionless_unscaled),
                         ('none', u.dimensionless_unscaled),
                         ('8=IMP8', u.dimensionless_unscaled),
+                        (' index value', u.dimensionless_unscaled),
+                        (' none', u.dimensionless_unscaled),
                         ('microW m^-2', u.mW * u.m**-2),
                         ('years', u.yr),
                         ('(2038=Yr0)', u.yr),
@@ -174,7 +176,9 @@ def cdf_dict(unit_string):
                         ('km/sec (from fits)', u.km / u.s),
                         ('km/sec (from moments)', u.km / u.s),
                         ('km (>200)', u.km),
-                        ('ionic charge', u.electron),
+                        ('ionic charge', u.dimensionless_unscaled),
+                        ('u/e', u.dimensionless_unscaled),
+                        ('Volts', u.V),
                         ('earth radii', u.earthRad),
                         ('Re', u.earthRad),
                         ('Earth Radii', u.earthRad),

diff --git a/heliopy/data/psp.py b/heliopy/data/psp.py
@@ -1,6 +1,7 @@
 """
 Methods for importing data from Parker Solar Probe.
 """
+import astropy.units as u
 import pathlib
 import urllib.error
 
@@ -37,6 +38,20 @@ class _SWEAPDownloader(_PSPDownloader):
 
 class _SWEAPL3Downloader(_SWEAPDownloader):
     badvalues = [-1e31]
+    units = {'u/e': u.dimensionless_unscaled}
+    # Fill in some missing units
+    for i in range(3):
+        for j in ['p', 'p1', 'a', '3']:
+            units[f'v{j}_fit_SC_{i}'] = u.km / u.s
+            units[f'v{j}_fit_SC_uncertainty_{i}'] = u.km / u.s
+            units[f'v{j}_fit_RTN_{i}'] = u.km / u.s
+            units[f'v{j}_fit_RTN_uncertainty_{i}'] = u.km / u.s
+            units[f'v{j}_moment_SC_{i}'] = u.km / u.s
+            units[f'v{j}_moment_SC_deltahigh_{i}'] = u.km / u.s
+            units[f'v{j}_moment_SC_deltalow_{i}'] = u.km / u.s
+            units[f'v{j}_moment_RTN_{i}'] = u.km / u.s
+            units[f'v{j}_moment_RTN_deltahigh_{i}'] = u.km / u.s
+            units[f'v{j}_moment_RTN_deltalow_{i}'] = u.km / u.s
 
     def local_dir(self, interval):
         year = interval.start.strftime('%Y')