def test_dump_1(self): """ generate node object for 1 story """ m = "This is a story" storylist_orig = dpcore.load_stories(m) nodelist = dpcore.yamlnodedump_stories(storylist_orig[0]) yml = yaml.serialize(nodelist) #print yml c = yaml.load(yml) storylist_comp = dpcore.load_stories(c) self.assertEqual(storylist_comp[0].story, storylist_orig[0].story) self.assertEqual(storylist_comp[0].note, storylist_orig[0].note)
def test_dump_2(self):
""" generate node object2 for 2 story """
m = ["This is story 1.", {"story": "This is story 2.", "sub-story": "This is a substory.\nwhich have 2 lines.", "task": "task 1.",}]
storylist_orig = dpcore.load_stories(m)
self.assertEqual(1, len(storylist_orig[1].substory))
self.assertEqual(1, len(storylist_orig[1].subtask))
nodeobjlist = dpcore.yamlnodedump_stories(storylist_orig)
nodelist = yaml.SequenceNode(tag=u"tag:yaml.org,2002:seq", value=nodeobjlist, flow_style=False)
yml = yaml.serialize(nodelist)
#print yml
c = yaml.load(yml)
storylist_comp = dpcore.load_stories(c)
for idx in range(2):
self.assertEqual(storylist_comp[idx].story, storylist_orig[idx].story)
self.assertEqual(storylist_comp[idx].note, storylist_orig[idx].note)
self.assertEqual(1, len(storylist_comp[1].substory))
self.assertEqual(1, len(storylist_comp[1].subtask))