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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
import unittest
import os
import shutil
from test_BaseCalculator import PlusCalculator, NumberData
from libpyvinyl.Instrument import Instrument
class InstrumentTest(unittest.TestCase):
"""
Test class for the Detector class.
"""
@classmethod
def setUpClass(cls):
"""Setting up the test class."""
input1 = NumberData.from_dict({"number": 1}, "input1")
input2 = NumberData.from_dict({"number": 1}, "input2")
calculator1 = PlusCalculator("test1", [input1, input2])
cls.calculator1 = calculator1
calculator2 = PlusCalculator("test2", [input1, input2])
calculator2.parameters["plus_times"] = 12
cls.calculator2 = calculator2
@classmethod
def tearDownClass(cls):
"""Tearing down the test class."""
pass
def setUp(self):
"""Setting up a test."""
self.__files_to_remove = []
self.__dirs_to_remove = []
def tearDown(self):
"""Tearing down a test."""
for f in self.__files_to_remove:
if os.path.isfile(f):
os.remove(f)
for d in self.__dirs_to_remove:
if os.path.isdir(d):
shutil.rmtree(d)
def testInstrumentConstruction(self):
"""Testing the default construction of the class."""
# Construct the object.
my_instrument = Instrument("myInstrument")
my_instrument.add_calculator(self.calculator1)
my_instrument.add_calculator(self.calculator2)
def testListCalculator(self):
"""Testing list calculators"""
# Construct the object.
my_instrument = Instrument("myInstrument")
my_instrument.add_calculator(self.calculator1)
my_instrument.add_calculator(self.calculator2)
my_instrument.list_calculators()
def testListParams(self):
"""Testing listing parameters"""
my_instrument = Instrument("myInstrument")
my_instrument.add_calculator(self.calculator1)
my_instrument.add_calculator(self.calculator2)
my_instrument.list_parameters()
def testRemoveCalculator(self):
"""Testing remove calculator"""
my_instrument = Instrument("myInstrument")
my_instrument.add_calculator(self.calculator1)
my_instrument.add_calculator(self.calculator2)
self.assertEqual(len(my_instrument.calculators), 2)
my_instrument.remove_calculator(self.calculator1.name)
self.assertEqual(len(my_instrument.calculators), 1)
def testEditCalculator(self):
"""Testing edit calculator"""
my_instrument = Instrument("myInstrument")
my_instrument.add_calculator(self.calculator1)
my_instrument.parameters["test1"]["plus_times"] = 10
my_instrument.parameters["test1"]["plus_times"] = 15
energy1 = my_instrument.calculators["test1"].parameters["plus_times"].value
self.assertEqual(energy1, 15)
def testAddMaster(self):
"""Testing remove calculator"""
my_instrument = Instrument("myInstrument")
my_instrument.add_calculator(self.calculator1)
my_instrument.add_calculator(self.calculator2)
links = {"test1": "plus_times", "test2": "plus_times"}
my_instrument.add_master_parameter("plus_times", links)
my_instrument.master["plus_times"] = 10
tims1 = my_instrument.calculators["test1"].parameters["plus_times"].value
tims2 = my_instrument.calculators["test2"].parameters["plus_times"].value
self.assertEqual(tims1, 10)
self.assertEqual(tims2, 10)
def testSetBasePath(self):
"""Testing setup base path for calculators"""
my_instrument = Instrument("myInstrument")
my_instrument.add_calculator(self.calculator1)
my_instrument.add_calculator(self.calculator2)
my_instrument.set_instrument_base_dir("test")
self.assertEqual(
my_instrument.calculators["test1"].base_dir, "test/PlusCalculator"
)
self.assertEqual(
my_instrument.calculators["test2"].base_dir, "test/PlusCalculator"
)
if __name__ == "__main__":
unittest.main()
|
