class OneDimensionCellExternalLibTest(unittest.TestCase):
"""
Unittest of the OneDimensionCell class (external lib use case)
"""
@classmethod
def setUpClass(cls):
"""
Tests setup for class
"""
data_file_path = os.path.join(os.path.dirname(__file__),
"../../../tests/0_UNITTEST/XDATA.json")
DataContainer(data_file_path)
@classmethod
def tearDownClass(cls):
DataContainer.clear()
print("\n ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ \n")
def setUp(self):
self.test_cell = Cell(3)
def tearDown(self):
pass
@unittest.skip("No yet ready")
def test_compute_new_pressure_external(self):
"""
Test of compute_new_pressure method with external solver
"""
self.test_cell._external_library = "" #Â pylint:disable=protected-access
self.test_cell.energy.current_value = np.array(
[1.e+06, 0.5e+05, 2.4e+07])
self.test_cell.pressure.current_value = np.array(
[1.5e+09, 0.5e+08, 1.2e+10])
self.test_cell.density.current_value = np.array([8000., 8500., 9500.])
self.test_cell.pseudo.current_value = np.array([1.e+08, 0., 2.4e+09])
self.test_cell.density.new_value = np.array([8120., 8440., 9620.])
self.test_cell.energy.new_value = np.array([0., 0., 0.])
self.test_cell.pressure.new_value = np.array([0., 0., 0.])
self.test_cell.sound_velocity.new_value = np.array([0., 0., 0.])
self.test_cell.cell_in_target = np.array([True, True, True])
self.test_cell.compute_new_pressure(np.array([True, True, True]),
1.e-6)
# Function to vanish
delta_v = 1. / self.test_cell.density.new_value - 1. / self.test_cell.density.current_value
func = (self.test_cell.energy.new_value +
self.test_cell.pressure.new_value * delta_v / 2. +
(self.test_cell.pressure.current_value +
2. * self.test_cell.pseudo.current_value) * delta_v / 2. -
self.test_cell.energy.current_value)
np.testing.assert_allclose(func, [0., 0., 0.])
class OneDimensionCellInternalLibTest(unittest.TestCase):
"""
A class to test the call of eos solver with internal lib
"""
@classmethod
def setUpClass(cls):
"""
Tests setup for class
"""
data_file_path = os.path.join(
os.path.dirname(__file__),
"../../../tests/0_UNITTEST/XDATA_hydro.json")
DataContainer(data_file_path)
@classmethod
def tearDownClass(cls):
DataContainer.clear()
print("\n ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ \n")
def setUp(self):
self.test_cell = Cell(3)
def tearDown(self):
pass
def test_compute_new_pressure_internal(self):
"""
Test of compute_new_pressure method with internal solver
"""
self.test_cell._external_library = None
self.test_cell.energy.current_value = np.array(
[1.e+06, 0.5e+05, 2.4e+07])
self.test_cell.pressure.current_value = np.array(
[1.5e+09, 0.5e+08, 1.2e+10])
self.test_cell.density.current_value = np.array([8000., 8500., 9500.])
self.test_cell.pseudo.current_value = np.array([1.e+08, 0., 2.4e+09])
self.test_cell.density.new_value = np.array([8120., 8440., 9620.])
self.test_cell.energy.new_value = np.array([0., 0., 0.])
self.test_cell.pressure.new_value = np.array([0., 0., 0.])
self.test_cell.sound_velocity.new_value = np.array([0., 0., 0.])
self.test_cell.compute_new_pressure(np.array([True, True, True]),
1.e-6)
# Function to vanish
delta_v = 1. / self.test_cell.density.new_value - 1. / self.test_cell.density.current_value
func = (self.test_cell.energy.new_value +
self.test_cell.pressure.new_value * delta_v / 2. +
(self.test_cell.pressure.current_value +
2. * self.test_cell.pseudo.current_value) * delta_v / 2. -
self.test_cell.energy.current_value)
np.testing.assert_allclose(
func, np.array([-1001570.197044, -49979.091163, -24011029.653135]))
class OneDimensionCellElastoTest(unittest.TestCase):
"""
A class to test the OneDimensionCell module with elasticity physic
"""
@classmethod
def setUpClass(cls):
"""
Tests setup for class
"""
data_file_path = os.path.join(os.path.dirname(__file__),
"../../../tests/0_UNITTEST/XDATA_elasto.json")
DataContainer(data_file_path)
@classmethod
def tearDownClass(cls):
DataContainer.clear()
print("\n ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ \n")
def setUp(self):
self.nbr_cells = 4
self.my_cells = OneDimensionCell(self.nbr_cells)
self.my_cells.cell_in_target = np.ones(self.nbr_cells, dtype='bool')
self.mask = np.array([True, True, False, False])
self.test_data = DataContainer() # pylint: disable=no-value-for-parameter
def tearDown(self):
pass
def test_compute_new_pressure_with_elasticity(self):
"""
Test de la m�thode compute_new_pressure
"""
mask_classic = np.array([True, True, False, False])
self.my_cells.density.current_value = np.ones(self.nbr_cells) * 8930.
self.my_cells.pressure.current_value = np.ones(self.nbr_cells) * 1.e+5
self.my_cells.energy.current_value = np.ones(self.nbr_cells) * 7.5
self.my_cells._deviatoric_stress_new = np.array([[200., -100, -100.],
[5., -2.5, -2.5],
[100., -50., -50.],
[400., -200., -200.]])
self.my_cells._deviatoric_strain_rate = \
np.array([[-0.235294, 0.117647, 0.117647],
[0.4444444, -0.2222222, -0.2222222],
[0., 0., 0.],
[0., 0., 0.]])
self.my_cells.density.new_value = np.array([8940., 8970, 8920., 9000.])
self.my_cells.sound_velocity.new_value = np.zeros([self.nbr_cells])
self.my_cells.pressure.new_value = np.zeros([self.nbr_cells])
self.my_cells.energy.new_value = np.zeros([self.nbr_cells])
self.my_cells.pseudo.new_value = np.zeros([self.nbr_cells])
delta_t = 1.
self.my_cells.compute_new_pressure(mask_classic, delta_t)
np.testing.assert_allclose(self.my_cells.pressure.new_value,
np.array([1.557157e+08, 6.266033e+08,
0.000000e+00, 0.000000e+00]), rtol=1.e-5)
np.testing.assert_allclose(self.my_cells.energy.new_value,
np.array([17.254756, 163.9763, 0., 0.]))
np.testing.assert_allclose(self.my_cells.sound_velocity.new_value,
np.array([3948.726929, 3974.84139, 0., 0.]))
def test_compute_shear_modulus(self):
"""
Test of the method compute_shear_modulus
"""
target_model = self.test_data.material_target
self.my_cells.compute_shear_modulus(
target_model.constitutive_model.elasticity_model.build_shear_modulus_obj(),
np.array([True, True, True, True]))
expected_value = target_model.initial_values.shear_modulus_init
np.testing.assert_allclose(self.my_cells.shear_modulus.new_value,
np.ones([self.nbr_cells]) * expected_value)
def test_compute_complete_stress_tensor(self):
"""
Test of the method compute_complete_stress_tensor
"""
self.my_cells._stress = np.array([[2000., -1000, -1000.],
[50., -25., -25.],
[1000., -500., -500.],
[4000., -2000., -2000.]])
self.my_cells._deviatoric_stress_new = np.array([[200., -100, -100.],
[5., -2.5, -2.5],
[100., -50., -50.],
[400., -200., -200.]])
self.my_cells.pressure.new_value = np.array([1000, 25, 500, 2000])
self.my_cells.pseudo.new_value = np.array([-1, -2, -3, 4])
self.my_cells.compute_complete_stress_tensor()
expected_result = np.array([[-799., -1099, -1099.],
[-18, -25.5, -25.5],
[-397., -547., -547.],
[-1604., -2204., -2204.]])
np.testing.assert_allclose(self.my_cells.stress, expected_result)
@mock.patch.object(OneDimensionCell, "general_method_deviator_strain_rate", spec=classmethod,
new_callable=mock.MagicMock)
def test_compute_deviatoric_stress_tensor(self, mock_compute_d):
"""
Test of the method compute_deviatoric_stress_tensor
"""
mock_compute_d.return_value = np.array([[1., -0.5, -0.5],
[3., -1.5, -1.5],
[2., -1., -1.],
[0., 0., 0.]])
mask = np.array([True, True, True, False])
delta_t = 1.
coord_noeud_new = np.array([[-0.25, ], [0.1, ], [0.2, ], [0.45, ], [0.85, ]])
vitesse_noeud_new = np.array([[0.1, ], [-0.05, ], [0., ], [0.2, ], [0.3, ]])
topo_ex = Topology1D(5, 4)
self.my_cells._deviatoric_stress_current = np.array([[2000., -1000, -1000.],
[50., -25., -25.],
[1000., -500., -500.],
[4000., -2000., -2000.]])
self.my_cells._deviatoric_strain_rate = np.array([[1., -0.5, -0.5],
[3., -1.5, -1.5],
[2., -1., -1.],
[1., -0.5, -0.5]])
self.my_cells.shear_modulus.new_value = np.array([2., 4., 6., 8.])
self.my_cells.compute_deviatoric_stress_tensor(mask, topo_ex, coord_noeud_new,
vitesse_noeud_new, delta_t)
np.testing.assert_allclose(self.my_cells._deviatoric_stress_new,
np.array([[2004, -1002., -1002.],
[74., -37., -37.],
[1024., -512., -512.],
[0., 0., 0.]]), rtol=1.e-05)
def test_compute_deviator_strain_rate(self):
"""
Test of the method compute_deviatoric_strain_rate
"""
# input data
mask = np.array([True, True, False, False])
delta_t = 1.
coord_noeud_new = np.array([[-0.25, ], [0.1, ], [0.2, ], [0.45, ], [0.85, ]])
vitesse_noeud_new = np.array([[0.1, ], [-0.05, ], [0., ], [0.2, ], [0.3, ]])
topo_ex = Topology1D(5, 4)
# Test of the method compute_deviator
D = self.my_cells.compute_deviator_strain_rate(delta_t, topo_ex, coord_noeud_new,
vitesse_noeud_new)
self.my_cells._deviatoric_strain_rate[mask] = D[mask]
np.testing.assert_allclose(self.my_cells._deviatoric_strain_rate,
np.array([[-0.235294, 0.117647, 0.117647],
[0.4444444, -0.2222222, -0.2222222],
[0., 0., 0.], [0., 0., 0.]]), rtol=1.e-05)