def test_HH_x_fit_single_layer(self):
data = np.genfromtxt(_join(f_dir, 'curve_FKSH14.txt'))
curve = Damping_Curve(data[:, 2:4])
try:
hhx = curve.get_HH_x_param(pop_size=1,
n_gen=1,
show_fig=True,
use_scipy=False)
self.assertEqual(len(hhx), 9)
self.assertEqual(
hhx.keys(), {
'gamma_t', 'a', 'gamma_ref', 'beta', 's', 'Gmax', 'mu',
'Tmax', 'd'
})
except ImportError: # DEAP library may not be installed
pass
hhx = curve.get_HH_x_param(pop_size=1,
n_gen=1,
show_fig=True,
use_scipy=True)
self.assertEqual(len(hhx), 9)
self.assertEqual(hhx.keys(), {
'gamma_t', 'a', 'gamma_ref', 'beta', 's', 'Gmax', 'mu', 'Tmax', 'd'
})
def test_H4_x_fit_single_layer(self):
data = np.genfromtxt('./files/curve_FKSH14.txt')
curve = Damping_Curve(data[:, 2:4])
try:
h4x = curve.get_H4_x_param(pop_size=1, n_gen=1, show_fig=True,
use_scipy=False)
self.assertEqual(len(h4x), 4)
self.assertEqual(h4x.keys(), {'gamma_ref', 's', 'beta', 'Gmax'})
except ImportError: # DEAP library may not be installed
pass
h4x = curve.get_H4_x_param(pop_size=1, n_gen=1, show_fig=True,
use_scipy=True)
self.assertEqual(len(h4x), 4)
self.assertEqual(h4x.keys(), {'gamma_ref', 's', 'beta', 'Gmax'})
def test_value_check(self):
data = np.genfromtxt('./files/curve_FKSH14.txt')[:, 2:4]
with self.assertRaises(ValueError, msg='G/Gmax values must be between'):
GGmax_Curve(data)
with self.assertRaises(ValueError, msg='damping values must be between'):
Damping_Curve(data * 100.0)
with self.assertRaises(ValueError, msg='should have all non-negative'):
Stress_Curve(data * -1)
def test_value_check(self):
data = np.genfromtxt(_join(f_dir, 'curve_FKSH14.txt'))[:, 2:4]
with self.assertRaisesRegex(ValueError,
'G/Gmax values must be between'):
GGmax_Curve(data)
with self.assertRaisesRegex(ValueError,
'damping values must be between'):
Damping_Curve(data * 100.0)
with self.assertRaisesRegex(ValueError,
'should have all non-negative'):
Stress_Curve(data * -1)
def test_multiple_damping_curves(self):
mdc = Multiple_Damping_Curves(_join(f_dir, 'curve_FKSH14.txt'))
# Test __len__
self.assertEqual(len(mdc), 5)
# Test __getitem__
strain_bench = [
0.0001, 0.0003, 0.001, 0.003, 0.01, 0.03, 0.1, 0.3, 1, 3
]
damping_bench = [
1.6683, 1.8386, 2.4095, 3.8574, 7.4976, 12.686, 18.102, 21.005,
21.783, 21.052
]
layer_0_bench = np.column_stack((strain_bench, damping_bench))
self.assertTrue(np.allclose(mdc[0].raw_data, layer_0_bench))
# Test __setitem__
with self.assertRaisesRegex(TypeError, 'new `item` must be of type'):
mdc[2] = 2.5 # use an incorrect type
mdc[4] = Damping_Curve(layer_0_bench)
self.assertTrue(np.allclose(mdc[0].raw_data, mdc[4].raw_data))
# Test __delitem__
mdc_2 = mdc[2]
mdc_3 = mdc[3]
del mdc[2]
self.assertEqual(len(mdc), 4)
self.assertEqual(mdc.n_layer, 4)
# Test __contains__
self.assertFalse(mdc_2 in mdc)
self.assertTrue(mdc_3 in mdc)
# Test slicing
mdc_slice = mdc[:2]
self.assertEqual(len(mdc_slice), 2)
self.assertTrue(isinstance(mdc_slice, Multiple_Damping_Curves))
self.assertTrue(isinstance(mdc_slice[0], Damping_Curve))
# Test append
with self.assertRaisesRegex(TypeError,
'`curve` should be a numpy array'):
mdc[2] = GGmax_Curve(_join(
f_dir, 'curve_FKSH14.txt')) # use an incorrect type
self.assertEqual(len(mdc), 4)
mdc.append(mdc_3)
self.assertEqual(len(mdc), 5)
self.assertEqual(mdc.n_layer, 5)