@@ -16,6 +16,38 @@ def test_load_file(self):
16
16
self .spec .loadFile (file_path , SpecUtils .ParserType .Auto )
17
17
self .assertTrue (self .spec .numMeasurements () > 0 )
18
18
19
+ def test_load_save_file (self ):
20
+ script_dir = Path (__file__ ).parent .resolve ()
21
+ file_path = os .path .join (script_dir , ".." , "examples" , "passthrough.n42" )
22
+
23
+ orig = SpecUtils .SpecFile ()
24
+ orig .loadFile (file_path , SpecUtils .ParserType .Auto )
25
+ self .assertTrue (orig .numMeasurements () > 0 )
26
+
27
+ pcffile = open ("passthrough_saved.pcf" , "wb" )
28
+ orig .writeToStream (pcffile , orig .sampleNumbers (), orig .detectorNames (), SpecUtils .SaveSpectrumAsType .Pcf )
29
+ pcffile .close ()
30
+
31
+ pcfreread = SpecUtils .SpecFile ()
32
+ pcfreread .loadFile ("passthrough_saved.pcf" , SpecUtils .ParserType .Pcf )
33
+ self .assertEqual (pcfreread .numMeasurements (), orig .numMeasurements ())
34
+
35
+ pcfinfile = open ("passthrough_saved.pcf" , "rb" )
36
+ pcfreread2 = SpecUtils .SpecFile ()
37
+ pcfreread2 .loadFromPcf (pcfinfile )
38
+ self .assertEqual (pcfreread2 .numMeasurements (), orig .numMeasurements ())
39
+ pcfinfile .close ()
40
+
41
+ n42outfile = open ("passthrough_saved.n42" , "wb" )
42
+ orig .writeToStream (n42outfile , orig .sampleNumbers (), orig .detectorNames (), SpecUtils .SaveSpectrumAsType .N42_2012 )
43
+ n42outfile .close ()
44
+
45
+ n42infile = open ("passthrough_saved.n42" , "rb" )
46
+ n42reread = SpecUtils .SpecFile ()
47
+ n42reread .loadFromN42 ( n42infile )
48
+ self .assertEqual (n42reread .numMeasurements (), orig .numMeasurements ())
49
+ n42infile .close ()
50
+
19
51
def test_create_measurement (self ):
20
52
# Create and configure a new measurement
21
53
meas = SpecUtils .Measurement ()
0 commit comments