36-pretty…

test_model_solver.py
View Options

			#!/usr/bin/env python3
			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')