1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import os
import h5py
import pytest
import numpy as np
from pynpoint.core.pypeline import Pypeline
from pynpoint.readwrite.textwriting import TextWritingModule
from pynpoint.util.tests import create_config, create_random, remove_test_data
class TestTextWriting:
def setup_class(self) -> None:
self.limit = 1e-10
self.test_dir = os.path.dirname(__file__) + "/"
create_random(self.test_dir, nimages=1)
create_config(self.test_dir + "PynPoint_config.ini")
self.pipeline = Pypeline(self.test_dir, self.test_dir, self.test_dir)
def teardown_class(self) -> None:
remove_test_data(self.test_dir, files=["image.dat", "data.dat"])
def test_input_data(self) -> None:
data = self.pipeline.get_data("images")
assert np.sum(data) == pytest.approx(
0.0008557279524431129, rel=self.limit, abs=0.0
)
assert data.shape == (1, 11, 11)
def test_text_writing(self) -> None:
text_write = TextWritingModule(
name_in="text_write",
data_tag="images",
file_name="image.dat",
output_dir=None,
header=None,
)
self.pipeline.add_module(text_write)
self.pipeline.run_module("text_write")
data = np.loadtxt(self.test_dir + "image.dat")
assert np.sum(data) == pytest.approx(
0.0008557279524431129, rel=self.limit, abs=0.0
)
assert data.shape == (11, 11)
def test_text_writing_ndim(self) -> None:
data_4d = np.random.normal(loc=0, scale=2e-4, size=(5, 5, 5, 5))
with h5py.File(self.test_dir + "PynPoint_database.hdf5", "a") as hdf_file:
hdf_file.create_dataset("data_4d", data=data_4d)
text_write = TextWritingModule(
name_in="write_4d",
data_tag="data_4d",
file_name="data.dat",
output_dir=None,
header=None,
)
self.pipeline.add_module(text_write)
with pytest.raises(ValueError) as error:
self.pipeline.run_module("write_4d")
assert str(error.value) == "Only 1D or 2D arrays can be written to a text file."
def test_text_writing_int(self) -> None:
data_int = np.arange(1, 11, 1)
with h5py.File(self.test_dir + "PynPoint_database.hdf5", "a") as hdf_file:
hdf_file.create_dataset("data_int", data=data_int)
text_write = TextWritingModule(
name_in="write_int",
data_tag="data_int",
file_name="data.dat",
output_dir=None,
header=None,
)
self.pipeline.add_module(text_write)
self.pipeline.run_module("write_int")
data = np.loadtxt(self.test_dir + "data.dat")
assert data == pytest.approx(data_int, rel=self.limit, abs=0.0)
assert data.shape == (10,)
|
