Diffusion tamaas (maxwell_viscoelastic-branch) Browse

test_dumper.py
99aeeb8af4edmaxwell_viscoelastic…

test_dumper.py
View Options

			# -- coding: utf-8 --
			# @file
			# LICENSE
			#
			# Copyright (©) 2016-2021 EPFL (École Polytechnique Fédérale de Lausanne),
			# Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
			#
			# This program is free software: you can redistribute it and/or modify
			# it under the terms of the GNU Affero General Public License as published
			# by the Free Software Foundation, either version 3 of the License, or
			# (at your option) any later version.
			#
			# This program 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 Affero General Public License for more details.
			#
			# You should have received a copy of the GNU Affero General Public License
			# along with this program. If not, see <https://www.gnu.org/licenses/>.

			from __future__ import division

			import os
			from pathlib import Path
			import tamaas as tm
			import numpy as np

			import pytest
			from conftest import mtidfn, type_list
			from tamaas.dumpers import NumpyDumper, JSONDumper, FieldDumper
			from tamaas.dumpers._helper import directory_dump, step_dump


			@pytest.fixture(scope="module",
			params=type_list,
			ids=mtidfn)
			def model_fixture(request):
			dim = tm.type_traits[request.param].dimension
			model = tm.ModelFactory.createModel(request.param, dim * [1.], dim * [4])
			model.displacement[:] = 3
			model.traction[:] = 1
			model['additional'] = 2 * np.ones(model.shape
			+ [tm.type_traits[model.type].components])
			return model


			class Dumper(tm.ModelDumper):
			"""Simple numpy dumper"""

			def __init__(self):
			tm.ModelDumper.__init__(self)

			def dump(self, model):
			np.savetxt('tractions.txt', np.ravel(model.traction))
			np.savetxt('displacement.txt', np.ravel(model.displacement))


			dumper_types = [NumpyDumper, JSONDumper]

			try:
			from tamaas.dumpers import H5Dumper
			dumper_types.append(H5Dumper)
			except ImportError:
			pass

			try:
			from tamaas.dumpers import UVWDumper
			dumper_types.append(UVWDumper)
			except ImportError:
			pass

			try:
			from tamaas.dumpers import NetCDFDumper
			dumper_types.append(NetCDFDumper)
			except ImportError:
			pass


			@pytest.mark.parametrize('dumper_class', dumper_types)
			def test_dumper(model_fixture, tmp_path, dumper_class):
			os.chdir(tmp_path)
			filename = 'test_{}'.format(dumper_class.__name__)

			if not issubclass(dumper_class, FieldDumper):
			dumper = dumper_class(filename)
			else:
			dumper = dumper_class(filename, all_fields=True)
			filename = dumper.file_path

			dumper << model_fixture

			# UVW does not read VTK files
			if dumper_class is UVWDumper:
			with pytest.raises(NotImplementedError):
			dumper.read(filename)
			assert Path(filename).is_file()
			return

			rmodel = dumper.read(filename)

			assert rmodel.type == model_fixture.type
			assert rmodel.shape == model_fixture.shape

			for field in model_fixture:
			# assert field in rmodel
			assert np.all(np.abs(rmodel[field] - model_fixture[field]) < 1e-15)


			def test_custom_dumper(tmp_path):
			os.chdir(tmp_path)
			model = tm.ModelFactory.createModel(tm.model_type.volume_2d, [1., 1., 1.],
			[16, 4, 8])
			dumper = Dumper()
			dumper << model

			tractions = np.loadtxt('tractions.txt')
			displacements = np.loadtxt('displacement.txt')

			assert np.all(tractions.reshape(model.traction.shape) == model.traction)
			assert np.all(displacements.reshape(model.displacement.shape)
			== model.displacement)


			@pytest.fixture
			def simple_model(scope="module"):
			return tm.ModelFactory.createModel(tm.model_type.basic_2d, [1, 1], [1, 1])


			def test_directory_dump(simple_model, tmp_path):
			os.chdir(tmp_path)

			@directory_dump("dummy")
			class Dummy(tm.ModelDumper):
			def __init__(self):
			super(Dummy, self).__init__()

			def dump(self, model):
			pass

			@property
			def file_path(self):
			return 'dummy'

			dumper = Dummy()
			dumper << simple_model

			assert Path('dummy').is_dir()


			def test_step_dump(simple_model, tmp_path):
			os.chdir(tmp_path)

			@step_dump
			class Dummy(FieldDumper):
			extension = 'dummy'

			def dump_to_file(self, file_descriptor, model):
			with open(file_descriptor, 'w'):
			pass

			files = []
			dumper = Dummy('dummy')

			def dump():
			files.append(dumper.file_path)
			dumper << simple_model

			dump()
			dump()

			for file in files:
			assert Path(file).is_file()

tamaas/tests/test_dumper.py99aeeb8af4edmaxwell_viscoelastic…

test_dumper.pyView Options

tamaas/tests/test_dumper.py
99aeeb8af4edmaxwell_viscoelastic…

test_dumper.py
View Options