def test_content_goes_to_named_file(self): fN = "node.yaml" fP = os.path.join(self.drcty.name, fN) self.assertFalse(os.path.isfile(fP)) with record(fN, parent=self.drcty.name) as output: output.write(ruamel.yaml.dump("Test string")) self.assertTrue(os.path.isfile(fP)) with open(fP, 'r') as check: self.assertEqual("Test string\n...", check.read().strip())
def test_content_goes_to_file_object(self): fObj = StringIO() with record(fObj, parent=self.drcty.name) as output: output.write(ruamel.yaml.dump("Test string")) self.assertEqual("Test string\n...", fObj.getvalue().strip())