forked from UXARRAY/uxarray
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
b8b8e17
commit 1ba6880
Showing
2 changed files
with
130 additions
and
108 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,130 @@ | ||
| import os | ||
| import numpy as np | ||
| import numpy.testing as nt | ||
| import random | ||
| import xarray as xr | ||
|
|
||
| from unittest import TestCase | ||
| from pathlib import Path | ||
|
|
||
| import uxarray as ux | ||
|
|
||
|
|
||
| from uxarray.grid.coordinates import node_lonlat_rad_to_xyz | ||
| from uxarray.grid.arcs import point_within_gca, in_between | ||
|
|
||
| try: | ||
| import constants | ||
| except ImportError: | ||
| from . import constants | ||
|
|
||
| # Data files | ||
| current_path = Path(os.path.dirname(os.path.realpath(__file__))) | ||
|
|
||
| class TestArcs(TestCase): | ||
|
|
||
| def test_pt_within_gcr(self): | ||
| # The GCR that's eexactly 180 degrees will have Value Error raised | ||
| gcr_180degree_cart = [ | ||
| ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, 0.0]), | ||
| ux.grid.coordinates.node_lonlat_rad_to_xyz([np.pi, 0.0]) | ||
| ] | ||
| pt_same_lon_in = ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, 0.0]) | ||
| with self.assertRaises(ValueError): | ||
| point_within_gca(pt_same_lon_in, gcr_180degree_cart) | ||
|
|
||
| gcr_180degree_cart = [ | ||
| ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, np.pi / 2.0]), | ||
| ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, -np.pi / 2.0]) | ||
| ] | ||
|
|
||
| pt_same_lon_in = ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, 0.0]) | ||
| with self.assertRaises(ValueError): | ||
| point_within_gca(pt_same_lon_in, gcr_180degree_cart) | ||
|
|
||
| # Test when the point and the GCA all have the same longitude | ||
| gcr_same_lon_cart = [ | ||
| ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, 1.5]), | ||
| ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, -1.5]) | ||
| ] | ||
| pt_same_lon_in = ux.grid.coordinates.node_lonlat_rad_to_xyz([0.0, 0.0]) | ||
| self.assertTrue(point_within_gca(pt_same_lon_in, gcr_same_lon_cart)) | ||
|
|
||
| pt_same_lon_out = ux.grid.coordinates.node_lonlat_rad_to_xyz( | ||
| [0.0, 1.500000000000001]) | ||
| res = point_within_gca(pt_same_lon_out, gcr_same_lon_cart) | ||
| self.assertFalse(res) | ||
|
|
||
| pt_same_lon_out_2 = ux.grid.coordinates.node_lonlat_rad_to_xyz( | ||
| [0.1, 1.0]) | ||
| res = point_within_gca(pt_same_lon_out_2, gcr_same_lon_cart) | ||
| self.assertFalse(res) | ||
|
|
||
| # And if we increase the digital place by one, it should be true again | ||
| pt_same_lon_out_add_one_place = ux.grid.coordinates.node_lonlat_rad_to_xyz( | ||
| [0.0, 1.5000000000000001]) | ||
| res = point_within_gca(pt_same_lon_out_add_one_place, gcr_same_lon_cart) | ||
| self.assertTrue(res) | ||
|
|
||
| # Normal case | ||
| # GCR vertex0 in radian : [1.3003315590159483, -0.007004587172323237], | ||
| # GCR vertex1 in radian : [3.5997458123873827, -1.4893379576608758] | ||
| # Point in radian : [1.3005410084914981, -0.010444274637648326] | ||
| gcr_cart_2 = np.array([[0.267, 0.963, -0.007], [-0.073, -0.036, | ||
| -0.997]]) | ||
| pt_cart_within = np.array( | ||
| [0.25616109352676675, 0.9246590335292105, -0.010021496695000144]) | ||
| self.assertTrue(point_within_gca(pt_cart_within, gcr_cart_2, True)) | ||
|
|
||
| # Test other more complicate cases : The anti-meridian case | ||
|
|
||
| # GCR vertex0 in radian : [5.163808182822441, 0.6351384888657234], | ||
| # GCR vertex1 in radian : [0.8280410325693055, 0.42237025187091526] | ||
| # Point in radian : [0.12574759138415173, 0.770098701904903] | ||
| gcr_cart = np.array([[0.351, -0.724, 0.593], [0.617, 0.672, 0.410]]) | ||
| pt_cart = np.array( | ||
| [0.9438777657502077, 0.1193199333436068, 0.922714737029319]) | ||
| self.assertTrue(point_within_gca(pt_cart, gcr_cart, is_directed=True)) | ||
| # If we swap the gcr, it should throw a value error since it's larger than 180 degree | ||
| gcr_cart_flip = np.array([[0.617, 0.672, 0.410], [0.351, -0.724, | ||
| 0.593]]) | ||
| with self.assertRaises(ValueError): | ||
| point_within_gca(pt_cart, gcr_cart_flip, is_directed=True) | ||
|
|
||
| # If we flip the gcr in the undirected mode, it should still work | ||
| self.assertTrue( | ||
| point_within_gca(pt_cart, gcr_cart_flip, is_directed=False)) | ||
|
|
||
| # 2nd anti-meridian case | ||
| # GCR vertex0 in radian : [4.104711496596806, 0.5352983676533828], | ||
| # GCR vertex1 in radian : [2.4269979227622533, -0.007003212877856825] | ||
| # Point in radian : [0.43400375562899113, -0.49554509841586936] | ||
| gcr_cart_1 = np.array([[-0.491, -0.706, 0.510], [-0.755, 0.655, | ||
| -0.007]]) | ||
| pt_cart_within = np.array( | ||
| [0.6136726305712109, 0.28442243941920053, -0.365605190899831]) | ||
| self.assertFalse( | ||
| point_within_gca(pt_cart_within, gcr_cart_1, is_directed=True)) | ||
| self.assertFalse( | ||
| point_within_gca(pt_cart_within, gcr_cart_1, is_directed=False)) | ||
|
|
||
| # The first case should not work and the second should work | ||
| v1_rad = [0.1, 0.0] | ||
| v2_rad = [2 * np.pi - 0.1, 0.0] | ||
| v1_cart = ux.grid.coordinates.node_lonlat_rad_to_xyz(v1_rad) | ||
| v2_cart = ux.grid.coordinates.node_lonlat_rad_to_xyz(v2_rad) | ||
| gcr_cart = np.array([v1_cart, v2_cart]) | ||
| pt_cart = ux.grid.coordinates.node_lonlat_rad_to_xyz([0.01, 0.0]) | ||
| with self.assertRaises(ValueError): | ||
| point_within_gca(pt_cart, gcr_cart, is_directed=True) | ||
| gcr_car_flipped = np.array([v2_cart, v1_cart]) | ||
| self.assertTrue( | ||
| point_within_gca(pt_cart, gcr_car_flipped, is_directed=True)) | ||
|
|
||
|
|
||
| class TestOperators(TestCase): | ||
|
|
||
| def test_in_between(self): | ||
| # Test the in_between operator | ||
| self.assertTrue(in_between(0, 1, 2)) | ||
| self.assertTrue(in_between(-1, -1.5, -2)) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters