Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F88621026
test_dumper.py
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sat, Oct 19, 19:28
Size
1 KB
Mime Type
text/x-python
Expires
Mon, Oct 21, 19:28 (2 d)
Engine
blob
Format
Raw Data
Handle
21799269
Attached To
rTAMAAS tamaas
test_dumper.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
print_function
,
division
import
tamaas
as
tm
import
numpy
as
np
import
os
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
.
getTraction
()))
np
.
savetxt
(
'displacement.txt'
,
np
.
ravel
(
model
.
getDisplacement
()))
def
test_dumper
(
tamaas_fixture
):
model
=
tm
.
ModelFactory
.
createModel
(
tm
.
model_type
.
volume_2d
,
[
1.
,
1.
,
1.
],
[
16
,
4
,
8
])
dumper
=
Dumper
()
model
.
setDumper
(
dumper
)
model
.
dump
()
tractions
=
np
.
loadtxt
(
'tractions.txt'
)
displacements
=
np
.
loadtxt
(
'displacement.txt'
)
assert
tractions
.
size
==
model
.
getTraction
()
.
size
assert
displacements
.
size
==
model
.
getDisplacement
()
.
size
for
name
in
[
'tractions.txt'
,
'displacement.txt'
]:
if
os
.
path
.
exists
(
name
):
os
.
remove
(
name
)
Event Timeline
Log In to Comment