@@ -16,6 +16,38 @@ def test_load_file(self):
1616 self .spec .loadFile (file_path , SpecUtils .ParserType .Auto )
1717 self .assertTrue (self .spec .numMeasurements () > 0 )
1818
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+
1951 def test_create_measurement (self ):
2052 # Create and configure a new measurement
2153 meas = SpecUtils .Measurement ()
0 commit comments