def test_get_discretized_profile(self):
Vs30 = 256
z1 = 100
svm = SVM(Vs30, z1=z1, show_fig=False)
# Test fixed_thk
discr_profile = svm.get_discretized_profile(fixed_thk=10,
show_fig=False)
self.assertTrue(isinstance(discr_profile, Vs_Profile))
if svm.has_bedrock_Vs: # bedrock Vs must match
self.assertTrue(svm.bedrock_Vs == discr_profile.vs_profile[-1, 1])
self.assertTrue(discr_profile.vs_profile[-1, 0] == 0)
# Test Vs_increment
discr_profile = svm.get_discretized_profile(Vs_increment=100,
show_fig=False)
self.assertTrue(isinstance(discr_profile, Vs_Profile))
if svm.has_bedrock_Vs: # bedrock Vs must match
self.assertTrue(svm.bedrock_Vs == discr_profile.vs_profile[-1, 1])
self.assertTrue(discr_profile.vs_profile[-1, 0] == 0)
# Test invalid Vs_increment
with self.assertRaises(ValueError, msg='max Vs of the smooth profile'):
svm.get_discretized_profile(Vs_increment=5000)
# Test input parameter checking
with self.assertRaises(ValueError, msg='You need to provide either'):
svm.get_discretized_profile(Vs_increment=None, fixed_thk=None)
with self.assertRaises(ValueError, msg='do not provide both'):
svm.get_discretized_profile(Vs_increment=1, fixed_thk=2)
def test_get_discretized_profile__valid_Vs_increment(self):
svm = SVM(target_Vs30=256, z1=100, show_fig=False)
discr_profile = svm.get_discretized_profile(Vs_increment=100,
show_fig=False)
self.assertTrue(isinstance(discr_profile, Vs_Profile))
if svm.has_bedrock_Vs: # bedrock Vs must match
self.assertTrue(svm.bedrock_Vs == discr_profile.vs_profile[-1, 1])
self.assertTrue(discr_profile.vs_profile[-1, 0] == 0)
def test_svm_profiles_plotting(self):
vs30 = 256 # m/s
z1 = 200 # m
svm = SVM(vs30, z1=z1, show_fig=False)
svm.get_discretized_profile(fixed_thk=20, show_fig=True)
svm.get_discretized_profile(Vs_increment=1, show_fig=True)
svm.get_discretized_profile(Vs_increment=100, show_fig=True)
svm.get_randomized_profile(show_fig=True)
def test_get_randomized_profile(self):
svm = SVM(target_Vs30=256, z1=100, show_fig=False)
random_profile = svm.get_randomized_profile(show_fig=False)
self.assertTrue(isinstance(random_profile, Vs_Profile))
if svm.has_bedrock_Vs: # bedrock Vs must match
self.assertTrue(svm.bedrock_Vs == random_profile.vs_profile[-1, 1])
self.assertTrue(random_profile.vs_profile[-1, 0] == 0)
# Use iteration to pick compliant randomized Vs profile
random_profile = svm.get_randomized_profile(show_fig=True,
vs30_z1_compliance=True,
verbose=True)
def test_Vs_cap(self):
Vs30 = 256
z1 = 10
# Case 1: Vs_cap is True (automatically chosen as 1,000 m/s)
svm = SVM(Vs30, z1=z1, Vs_cap=True)
self.assertEqual(svm.base_profile.vs_profile[-1, 0], 0)
self.assertEqual(svm.base_profile.vs_profile[-1, 1], 1000)
# Case 2: Vs_cap is user-defined
svm = SVM(Vs30, z1=z1, Vs_cap=1234.5)
self.assertEqual(svm.base_profile.vs_profile[-1, 0], 0)
self.assertEqual(svm.base_profile.vs_profile[-1, 1], 1234.5)
def test_index_closest(self):
array = [0, 1, 2, 1.1, 0.4, -3.2]
i, val = SVM._find_index_closest(array, 2.1)
self.assertEqual((i, val), (2, 2))
i, val = SVM._find_index_closest(array, -9)
self.assertEqual((i, val), (5, -3.2))
i, val = SVM._find_index_closest(array, -0.5)
self.assertEqual((i, val), (0, 0))
i, val = SVM._find_index_closest([1], 10000)
self.assertEqual((i, val), (0, 1))
with self.assertRaisesRegex(ValueError, 'length of `array` needs to'):
SVM._find_index_closest([], 2)
def test_get_discretized_profile__both_input_param_are_provided(self):
svm = SVM(target_Vs30=256, z1=100, show_fig=False)
with self.assertRaisesRegex(ValueError, 'do not provide both'):
svm.get_discretized_profile(Vs_increment=1, fixed_thk=2)
def test_get_discretized_profile__both_input_param_are_None(self):
svm = SVM(target_Vs30=256, z1=100, show_fig=False)
with self.assertRaisesRegex(ValueError, 'You need to provide either'):
svm.get_discretized_profile(Vs_increment=None, fixed_thk=None)
def test_get_discretized_profile__invalid_Vs_increment(self):
svm = SVM(target_Vs30=256, z1=100, show_fig=False)
with self.assertRaisesRegex(ValueError,
'max Vs of the smooth profile'):
svm.get_discretized_profile(Vs_increment=5000)
def test_base_profile(self):
svm = SVM(target_Vs30=256, z1=100, show_fig=False)
base_profile = svm.base_profile
self.assertTrue(isinstance(base_profile, Vs_Profile))
def test_Vs_cap_is_user_defined(self):
Vs30 = 256
z1 = 10
svm = SVM(Vs30, z1=z1, Vs_cap=1234.5)
self.assertEqual(svm.base_profile.vs_profile[-1, 0], 0)
self.assertEqual(svm.base_profile.vs_profile[-1, 1], 1234.5)
def test_Vs_cap_is_True(self):
Vs30 = 256
z1 = 10
svm = SVM(Vs30, z1=z1, Vs_cap=True)
self.assertEqual(svm.base_profile.vs_profile[-1, 0], 0)
self.assertEqual(svm.base_profile.vs_profile[-1, 1], 1000)
def test_init(self):
Vs30 = 256
z1 = 100
svm = SVM(target_Vs30=256, z1=100, show_fig=False)
self.assertEqual(svm.Vs30, Vs30)
self.assertEqual(svm.z1, z1)
def test_base_profile(self):
Vs30 = 256
z1 = 100
svm = SVM(Vs30, z1=z1, show_fig=False)
base_profile = svm.base_profile
self.assertTrue(isinstance(base_profile, Vs_Profile))