test_model_solver.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	#!/usr/bin/env python3

	# ------------------------------------------------------------------------------
	__author__ = "Nicolas Richart"
	__copyright__ = "Copyright (C) 2016-2018, EPFL (Ecole Polytechnique Fédérale" \
	" de Lausanne) Laboratory (LSMS - Laboratoire de Simulation" \
	" en Mécanique des Solides)"
	__credits__ = ["Nicolas Richart"]
	__license__ = "L-GPLv3"
	__maintainer__ = "Nicolas Richart"
	__email__ = "nicolas.richart@epfl.ch"
	# ------------------------------------------------------------------------------

	import numpy as np
	import numpy.linalg as npl
	import python_fe as pfe

	sim_u = np.genfromtxt("disp.csv", delimiter=",", names=True)

	Ne = int(sim_u['node'][-1])
	F = sim_u['force']
	blocked = sim_u['blocked']
	u = sim_u['disp']

	trusses = pfe.TrussFE(Ne=Ne,
	F={f: [np.where(F == f)] for f in np.unique(F)},
	blocked=(np.where(blocked == 1),
	u[blocked == 1]))

	solver = pfe.StaticSolver(trusses)
	solver.solveStep()

	upy = trusses.u

	n = npl.norm(upy - u) / npl.norm(upy)

	if n > 1e-14:
	raise ValueError('Something went wrong in the simulation')