def test_experimentlist_dumper_dump_formats(dials_regression, tmpdir): tmpdir.chdir() os.environ['DIALS_REGRESSION'] = dials_regression # Get all the filenames filename1 = os.path.join(dials_regression, 'experiment_test_data', 'experiment_1.json') # Read all the experiment lists in elist1 = ExperimentListFactory.from_json_file(filename1) # Create the experiment list dumper dump = ExperimentListDumper(elist1) # Dump as JSON file and reload filename = 'temp1.json' dump.as_json(filename) elist2 = ExperimentListFactory.from_json_file(filename) check(elist1, elist2) # Dump as split JSON file and reload filename = 'temp2.json' dump.as_json(filename, split=True) elist2 = ExperimentListFactory.from_json_file(filename) check(elist1, elist2) # Dump as pickle and reload filename = 'temp.pickle' dump.as_pickle(filename) elist2 = ExperimentListFactory.from_pickle_file(filename) check(elist1, elist2)
def tst_dump_formats(self): from uuid import uuid4 from os.path import join import os os.environ['DIALS_REGRESSION'] = self.path # Get all the filenames filename1 = join(self.path, 'experiment_test_data', 'experiment_1.json') # Read all the experiment lists in elist1 = ExperimentListFactory.from_json_file(filename1) # Create the experiment list dumper dump = ExperimentListDumper(elist1) # Dump as JSON file and reload filename = 'temp%s.json' % uuid4().hex dump.as_json(filename) elist2 = ExperimentListFactory.from_json_file(filename) self.check(elist1, elist2) # Dump as split JSON file and reload filename = 'temp%s.json' % uuid4().hex dump.as_json(filename, split=True) elist2 = ExperimentListFactory.from_json_file(filename) self.check(elist1, elist2) # Dump as pickle and reload filename = 'temp%s.pickle' % uuid4().hex dump.as_pickle(filename) elist2 = ExperimentListFactory.from_pickle_file(filename) self.check(elist1, elist2)