def test_no_BC(self):
#expected is from:
#np.dot((np.cos(z1[:,np.newaxis] * m[np.newaxis,:])-np.cos(z2[:,np.newaxis] * m[np.newaxis,:]))/m[np.newaxis,:]/(z2[:,np.newaxis]-z1[:,np.newaxis]), v_E_Igamv_the)
assert_allclose(dim1sin_avgf(self.m,
self.outz,
self.tvals,
self.v_E_Igamv_the,
drn=0),
np.array([[ 1.6275434 , 1.6275434 ],
[ 2.29709903, 2.29709903]]))
def test_bot_vs_time_drn_0(self):
#expected is from:
#test no_bc + bot_vs time interplolated at tvals and depth z (mag_vs_depth reduces to zero at top)
assert_allclose(dim1sin_avgf(self.m,
self.outz,
self.tvals,
self.v_E_Igamv_the,
drn=0,
bot_vs_time = [self.bot_vs_time]),
np.array([[ 1*0.3+1.6275434 , 2*0.3+1.6275434 ],
[ 1*0.5+2.29709903, 2*0.5+2.29709903]]))
def test_top_vs_time_drn_1(self):
#expected is from:
#test no_bc + top_vs time interplolated at tvals and depth z (mag vs depth is uniform top to bottom)
assert_allclose(dim1sin_avgf(self.m,
self.outz,
self.tvals,
self.v_E_Igamv_the,
drn=1,
top_vs_time = [self.top_vs_time]),
np.array([[ 1+1.6275434 , 2+1.6275434 ],
[ 1+2.29709903, 2+2.29709903]]))
def test_bot_vs_time_top_vs_time_drn_1_double_loads(self):
#expected is from:
#test no_bc + top_vs_time and bot_vs time interplolated at tvals and depth z (mag_vs_depth is uniform from top plus reduces to zero at top)
assert_allclose(dim1sin_avgf(self.m,
self.outz,
self.tvals,
self.v_E_Igamv_the,
drn=1,
top_vs_time = [self.top_vs_time, self.top_vs_time],
bot_vs_time = [self.bot_vs_time, self.bot_vs_time]),
np.array([[ 2+2*0.3+1.6275434 , 4+4*0.3+1.6275434 ],
[ 2+2*0.5+2.29709903, 4+4*0.5+2.29709903]]))
def test_top_vs_time_drn_0_omega_phase(self):
#expected is from:
#test no_bc + cos(omega*t+phase) multiplied top_vs time interplolated at tvals and depth z (mag_vs_depth reduces to zero at bottom)
assert_allclose(dim1sin_avgf(self.m,
self.outz,
self.tvals,
self.v_E_Igamv_the,
drn=0,
top_vs_time = [self.top_vs_time],
top_omega_phase=[self.omega_phase]),
np.array([[ 1*0.7*np.cos(1*1+2)+1.6275434, 2*0.7*np.cos(1*3+2)+1.6275434],
[ 1*0.5*np.cos(1*1+2)+2.29709903, 2*0.5*np.cos(1*3+2)+2.29709903]]))