Skip to content
Snippets Groups Projects
Commit 23903d9d authored by Mads M. Pedersen's avatar Mads M. Pedersen
Browse files

fixed test

parent b233dd6d
No related branches found
No related tags found
No related merge requests found
Pipeline #
Hide whitespace changes
Inline Side-by-side
wetb/wind/turbulence/tests/test_constraint_file.py
+ 13
− 13
View file @ 23903d9d
...@@ -19,18 +19,18 @@ class TestConstraintFile(unittest.TestCase): ...@@ -19,18 +19,18 @@ class TestConstraintFile(unittest.TestCase):
tu = np.array([time, u]).T tu = np.array([time, u]).T
tuvw = np.array([time, u, v, w]).T tuvw = np.array([time, u, v, w]).T
glpos = (0, 0, -85) glpos = (0, 0, -85)
constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints(glpos, tu) constraint_file.add_constraints(glpos, tu)
np.testing.assert_array_equal(constraint_file.constraints['u'][1], (6, 4, 4, 0)) np.testing.assert_array_equal(constraint_file.constraints['u'][1], (6, 4, 4, 0))
constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, -90), box_transport_speed=5, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, -90), box_transport_speed=5, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints(glpos, tuvw) constraint_file.add_constraints(glpos, tuvw)
np.testing.assert_array_equal(constraint_file.constraints['u'][2], (6, 4, 4, 2)) np.testing.assert_array_equal(constraint_file.constraints['u'][2], (6, 4, 4, 2))
np.testing.assert_array_equal(constraint_file.constraints['v'][2], (6, 4, 4, -.5)) np.testing.assert_array_equal(constraint_file.constraints['v'][2], (6, 4, 4, -.5))
np.testing.assert_array_equal(constraint_file.constraints['w'][2], (6, 4, 4, 2)) np.testing.assert_array_equal(constraint_file.constraints['w'][2], (6, 4, 4, 2))
constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints((-10, -4, -95), tu) constraint_file.add_constraints((-10, -4, -95), tu)
np.testing.assert_array_equal(constraint_file.constraints['u'][1], (8, 5, 5, 0)) np.testing.assert_array_equal(constraint_file.constraints['u'][1], (8, 5, 5, 0))
constraint_file = ConstraintFile(center_gl_xyz=(-5, 4, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(-5, 4, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints((-10, 0, -95), tu) constraint_file.add_constraints((-10, 0, -95), tu)
np.testing.assert_array_equal(constraint_file.constraints['u'][1], (8, 5, 5, 0)) np.testing.assert_array_equal(constraint_file.constraints['u'][1], (8, 5, 5, 0))
#constraint_file.save("./test_files", "constraints_test",'') #constraint_file.save("./test_files", "constraints_test",'')
...@@ -41,7 +41,7 @@ class TestConstraintFile(unittest.TestCase): ...@@ -41,7 +41,7 @@ class TestConstraintFile(unittest.TestCase):
tu = np.array([time, u]).T tu = np.array([time, u]).T
glpos = (0, 0, -85) glpos = (0, 0, -85)
constraint_file = ConstraintFile(center_gl_xyz=(0, 0, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(0, 0, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints([0,0,0], tu, subtract_mean=False) constraint_file.add_constraints([0,0,0], tu, subtract_mean=False)
np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 5, 5, 5)) np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 5, 5, 5))
constraint_file.add_constraints([5,0,0], tu, subtract_mean=False) constraint_file.add_constraints([5,0,0], tu, subtract_mean=False)
...@@ -56,19 +56,19 @@ class TestConstraintFile(unittest.TestCase): ...@@ -56,19 +56,19 @@ class TestConstraintFile(unittest.TestCase):
u = [5] u = [5]
tu = np.array([time, u]).T tu = np.array([time, u]).T
constraint_file = ConstraintFile(center_gl_xyz=(0, 0, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(0, 0, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints([0,0,0], tu, subtract_mean=False) constraint_file.add_constraints([0,0,0], tu, subtract_mean=False)
np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 5, 5, 5)) np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 5, 5, 5))
constraint_file = ConstraintFile(center_gl_xyz=(0, 0, -5), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(0, 0, -5), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints([0,0,0], tu, subtract_mean=False) constraint_file.add_constraints([0,0,0], tu, subtract_mean=False)
np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 5, 4, 5)) np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 5, 4, 5))
constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(-5, 0, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints([0,0,0], tu, subtract_mean=False) constraint_file.add_constraints([0,0,0], tu, subtract_mean=False)
np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 4, 5, 5)) np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (1, 4, 5, 5))
constraint_file = ConstraintFile(center_gl_xyz=(0, 10, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(0, 10, 0), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.add_constraints([0,0,0], tu, subtract_mean=False) constraint_file.add_constraints([0,0,0], tu, subtract_mean=False)
np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (6, 5, 5, 5)) np.testing.assert_array_equal(constraint_file.constraints['u'][-1], (6, 5, 5, 5))
...@@ -81,7 +81,7 @@ class TestConstraintFile(unittest.TestCase): ...@@ -81,7 +81,7 @@ class TestConstraintFile(unittest.TestCase):
tu = np.array([time, u]).T tu = np.array([time, u]).T
tuvw = np.array([time, u, v, w]).T tuvw = np.array([time, u, v, w]).T
glpos = (0, 0, -85) glpos = (0, 0, -85)
constraint_file = ConstraintFile(center_gl_xyz=(0, 0, -85), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(0, 0, -85), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
self.assertRaisesRegex(ValueError, "At time, t=0, global position \(0,0,-44\)", constraint_file.add_constraints, (0, 0, -85 + 35 + 6), tu) self.assertRaisesRegex(ValueError, "At time, t=0, global position \(0,0,-44\)", constraint_file.add_constraints, (0, 0, -85 + 35 + 6), tu)
self.assertRaisesRegex(ValueError, "At time, t=0, global position \(0,0,-126\)", constraint_file.add_constraints, (0, 0, -85 - 35 - 6), tu) self.assertRaisesRegex(ValueError, "At time, t=0, global position \(0,0,-126\)", constraint_file.add_constraints, (0, 0, -85 - 35 - 6), tu)
self.assertRaisesRegex(ValueError, "At time, t=0, global position \(-41,0,-85\)", constraint_file.add_constraints, (-35 - 6, 0, -85), tu) self.assertRaisesRegex(ValueError, "At time, t=0, global position \(-41,0,-85\)", constraint_file.add_constraints, (-35 - 6, 0, -85), tu)
...@@ -90,18 +90,18 @@ class TestConstraintFile(unittest.TestCase): ...@@ -90,18 +90,18 @@ class TestConstraintFile(unittest.TestCase):
self.assertRaisesRegex(ValueError, "At time, t=2, global position \(0,-13,-85\)", constraint_file.add_constraints, (0, -13, -85), tu) self.assertRaisesRegex(ValueError, "At time, t=2, global position \(0,-13,-85\)", constraint_file.add_constraints, (0, -13, -85), tu)
def test_hawc2_cmd(self): def test_hawc2_cmd(self):
constraint_file = ConstraintFile(center_gl_xyz=(0, 0, -85), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(0, 0, -85), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
mann = constraint_file.hawc2_mann_section("test",1) mann = constraint_file.hawc2_mann_section("test",1)
self.assertEqual(mann.box_dim_u.values, [16,1.875]) self.assertEqual(mann.box_dim_u.values, [16,1.875])
def test_load(self): def test_load(self):
constraint_file = ConstraintFile(center_gl_xyz=(-5, 4, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(-5, 4, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.load(get_test_file('constraints_test.con')) constraint_file.load(get_test_file('constraints_test.con'))
with open(get_test_file('constraints_test.con')) as fid: with open(get_test_file('constraints_test.con')) as fid:
self.assertEqual(str(constraint_file), fid.read()) self.assertEqual(str(constraint_file), fid.read())
def test_time_series(self): def test_time_series(self):
constraint_file = ConstraintFile(center_gl_xyz=(-5, 4, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_dimension=(30, 70, 70)) constraint_file = ConstraintFile(center_gl_xyz=(-5, 4, -90), box_transport_speed=10, no_grid_points=(16, 8, 8), box_size=(30, 70, 70))
constraint_file.load(get_test_file('constraints_test.con')) constraint_file.load(get_test_file('constraints_test.con'))
time,uvw = constraint_file.time_series(-4) time,uvw = constraint_file.time_series(-4)
u,v,w = uvw.T u,v,w = uvw.T
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment