Homec4science
Diffusion tamaas (ci-fix-static-checks) Browse

tamaas/tests/test_hertz.py
c6be98825283ci-fix-static-checks

Last Change
Raw File

test_hertz.py
View Options

#!/usr/bin/env python
# coding: utf-8
# -----------------------------------------------------------------------------
# @author Lucas Frérot <lucas.frerot@epfl.ch>
#
# @section LICENSE
#
# Copyright (©) 2016 EPFL (Ecole Polytechnique Fédérale de
# Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des
# Solides)
#
# Tamaas is free software: you can redistribute it and/or modify it under the
# terms of the GNU Lesser General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option) any
# later version.
#
# Tamaas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with Tamaas. If not, see <http://www.gnu.org/licenses/>.
# -----------------------------------------------------------------------------
from __future__ import division, print_function
import numpy as np
from numpy.linalg import norm
import tamaas as tm
import pytest
@pytest.fixture(scope="module",
params=[tm.PolonskyKeerRey.pressure])
def pkr(hertz, request):
model = tm.ModelFactory.createModel(tm.model_type.basic_2d,
[hertz.domain_size, hertz.domain_size],
[hertz.n, hertz.n])
model.setElasticity(hertz.e_star, 0)
solver = tm.PolonskyKeerRey(model, hertz.surface, 1e-12,
request.param,
request.param)
solver.solve(hertz.load)
return model, hertz
def test_contact_area(pkr):
# Testing contact area against Hertz solution for solids of revolution
model, hertz = pkr
contact_area = np.sum(model.getTraction() > 0) / float(hertz.n**2)
hertz_area = np.sum(hertz.pressure > 0) / float(hertz.n**2)
area_error = np.abs(hertz_area - contact_area) / hertz_area
assert area_error < 1e-2
def test_pressure(pkr):
model, hertz = pkr
error = norm(model.getTraction() - hertz.pressure) / norm(hertz.pressure)
assert error < 5e-3
def test_displacement(pkr):
model, hertz = pkr
error = norm(model.getDisplacement() - hertz.displacement) / norm(hertz.displacement)
assert error < 8e-3
if __name__ == "__main__":
print("Test meant to be run with pytest")
c4science · Help