diff --git a/wetb/utils/geometry.py b/wetb/utils/geometry.py index 211a811439af5fba0a05608dcfdc433b1a6cca56..a50b18be446662703d07c26ffb40393e86cd9fad 100644 --- a/wetb/utils/geometry.py +++ b/wetb/utils/geometry.py @@ -88,4 +88,8 @@ def std_rad(dir): return np.sqrt(1 - (np.nanmean(np.sin(dir)) ** 2 + np.nanmean(np.cos(dir)) ** 2)) def rpm2rads(rpm): - return rpm * 2 * np.pi / 60 \ No newline at end of file + return rpm * 2 * np.pi / 60 + + +def rads2rpm(rads): + return rads/ (2 * np.pi) * 60 \ No newline at end of file diff --git a/wetb/utils/tests/test_geometry.py b/wetb/utils/tests/test_geometry.py index 581dc34911335b5facd483cfd79b8cc2e739361c..db2f099d956f3a712b87c53b7391e2a6e345099a 100644 --- a/wetb/utils/tests/test_geometry.py +++ b/wetb/utils/tests/test_geometry.py @@ -13,7 +13,8 @@ import unittest import numpy as np -from wetb.utils.geometry import rad, deg, mean_deg, sind, cosd, std_deg, tand +from wetb.utils.geometry import rad, deg, mean_deg, sind, cosd, std_deg, tand,\ + rpm2rads, rads2rpm import os @@ -67,6 +68,10 @@ class TestGeometry(unittest.TestCase): def test_std_deg_nan(self): self.assertAlmostEqual(std_deg(np.array([0, 90, 180, 270, np.nan])), 57.296, 2) + + def test_rpm2rads(self): + self.assertAlmostEqual(rpm2rads(1),.1047,4) + self.assertAlmostEqual(rads2rpm(rpm2rads(1)), 1) if __name__ == "__main__":