def save_and_compare(arbor):
"""
Check that arbor saves correctly.
"""
fn = arbor.save_arbor()
save_arbor = load(fn)
assert isinstance(save_arbor, YTreeArbor)
compare_arbors(save_arbor, arbor)
def arbor(self):
if self._arbor is None:
if not os.path.exists(self.test_filename):
test_filename = \
os.path.join(test_data_dir, self.test_filename)
if os.path.exists(test_filename):
self.test_filename = test_filename
else:
self.skipTest("test file missing")
self._arbor = load(self.test_filename)
return self._arbor