def test_section_item_remove(self): """ Remove section item """ text = "\n".join([self.start, "[section]\nx = 3\ny = 7\n", self.end]) field = StructuredField(text) field.remove("section", "x") self.assertEqual( field.save(), "\n".join([self.start, "[section]\ny = 7\n", self.end]))
def test_multiple_values(self): """ Multiple values """ # Reading multiple values section = "[section]\nkey=val1 # comment\nkey = val2\n key = val3 " text = "\n".join([self.start, section, self.end]) field = StructuredField(text, multi=True) self.assertEqual(field.get("section", "key"), ["val1", "val2", "val3"]) # Writing multiple values values = ['1', '2', '3'] field = StructuredField(multi=True) field.set("section", values, "key") self.assertEqual(field.get("section", "key"), values) self.assertTrue("key = 1\nkey = 2\nkey = 3" in field.save()) # Remove multiple values field.remove("section", "key") self.assertTrue("key = 1\nkey = 2\nkey = 3" not in field.save()) self.assertRaises(StructuredFieldError, field.get, "section", "key")