diff --git a/pylibfdt/libfdt.i b/pylibfdt/libfdt.i index 51db953..fa5f499 100644 --- a/pylibfdt/libfdt.i +++ b/pylibfdt/libfdt.i @@ -908,6 +908,12 @@ class Property(bytearray): bytearray.__init__(self, value) self.name = name + def __copy__(self): + return type(self)(self.name, self) + + def __deepcopy__(self, memo): + return type(self)(self.name, self) + def as_cell(self, fmt): return struct.unpack('>' + fmt, self)[0] diff --git a/tests/pylibfdt_tests.py b/tests/pylibfdt_tests.py index 1873096..ae46b29 100644 --- a/tests/pylibfdt_tests.py +++ b/tests/pylibfdt_tests.py @@ -4,6 +4,7 @@ # Written by Simon Glass # +import copy import struct import sys import types @@ -192,6 +193,11 @@ class PyLibfdtBasicTests(unittest.TestCase): prop = self.fdt.get_property_by_offset(poffset) self.assertEqual(prop.name, 'compatible') self.assertEqual(prop, b'test_tree1\0') + for copied in (copy.copy(prop), copy.deepcopy(prop)): + self.assertIsInstance(copied, libfdt.Property) + self.assertIsNot(copied, prop) + self.assertEqual(copied.name, 'compatible') + self.assertEqual(copied, b'test_tree1\0') with self.assertRaises(FdtException) as e: self.fdt.get_property_by_offset(-2)