add copy and deepcopy dunder methods

Author: phlptp

python/units_python.cpp

@@ -85,6 +85,14 @@ NB_MODULE(units_llnl_ext, mod)
             "multiplier"_a,
             "unit"_a,
             "commodity"_a)
+        .def("__copy__", [](const units::precise_unit* unit)
+            {
+                return units::precise_unit(*unit);
+            })
+        .def("__deepcopy__", [](const units::precise_unit* unit, nb::dict & /*memo*/)
+            {
+                return units::precise_unit(*unit);
+            })
         .def_prop_ro("multiplier", &units::precise_unit::multiplier)
         .def_prop_ro(
             "commodity",
@@ -372,6 +380,14 @@ NB_MODULE(units_llnl_ext, mod)
             },
             "value"_a,
             "unit"_a)
+        .def("__copy__", [](const units::precise_measurement& copy)
+            {
+                return units::precise_measurement(copy);
+            })
+        .def("__deepcopy__", [](const units::precise_measurement& copy, nb::dict & /*memo*/)
+            {
+                return units::precise_measurement(copy);
+            })
         .def_prop_ro(
             "value",
             [](const units::precise_measurement& measurement) {
@@ -718,6 +734,14 @@ NB_MODULE(units_llnl_ext, mod)
                 new (dim) Dimension{def};
             },
             "composition"_a)
+        .def("__copy__", [](const Dimension& copy)
+            {
+                return Dimension(copy);
+            })
+        .def("__deepcopy__", [](const Dimension* dim, nb::dict & /*memo*/)
+            {
+                return Dimension(*dim);
+            })
         .def_prop_ro(
             "default_unit",
             [](const Dimension& dim) { return (dim.base); },

test/python/test_dimensions.py

@@ -7,6 +7,7 @@
 
 from units_llnl import Unit, Measurement, Dimension, asdimension
 
+import copy
 
 def test_dimensions():
     d1 = Dimension("length")
@@ -87,3 +88,27 @@ def test_dimensions_decompose2():
     dim2 = Dimension(decomposition)
     assert dim2 == dim
     assert dim2.default_unit == u1
+
+
+def test_copy():
+    m1=u.Dimension("10 m")
+    m2=copy.copy(m1)
+    assert m1==m2
+    
+def test_deepcopy():
+    m1=u.Dimension("10 m")
+    m2=copy.deepcopy(m1)
+    assert m1==m2
+    
+def test_deepcopy_object():
+    class val1:
+        value1=u.Dimension("10 m")
+        value2=u.Dimension("20 kg")
+        value3=u.Dimension("30 A")
+        values=[value1,value2,value3]
+        
+    v=val1()
+    v2=copy.deepcopy(v)
+    assert v2.value1==v.value1 
+    v.value1=u.Dimension("50 mol")
+    assert v2.value1==u.Dimension("10 m")

test/python/test_measurement.py

@@ -5,7 +5,7 @@
 
 import units_llnl as u
 import math
-
+import copy
 
 def test_basic_measurement():
     m1 = u.Measurement("10 m")
@@ -315,3 +315,27 @@ def test_to_dict():
     dv = m1.to_dict()
     assert dv["value"] == 25
     assert dv["unit"] == "lb"
+
+
+def test_copy():
+    m1=u.Measurement("10 m")
+    m2=copy.copy(m1)
+    assert m1==m2
+    
+def test_deepcopy():
+    m1=u.Measurement("10 m")
+    m2=copy.deepcopy(m1)
+    assert m1==m2
+    
+def test_deepcopy_object():
+    class val1:
+        value1=u.Measurement("10 m")
+        value2=u.Measurement("20 m")
+        value3=u.Measurement("30 m")
+        values=[value1,value2,value3]
+        
+    v=val1()
+    v2=copy.deepcopy(v)
+    assert v2.value1==v.value1 
+    v.value1=u.Measurement("50 m")
+    assert v2.value1==u.Measurement("10 m")

test/python/test_units.py

@@ -4,7 +4,7 @@
 # SPDX-License-Identifier: BSD-3-Clause
 
 import units_llnl as u
-
+import copy
 
 def test_basic_unit():
     u1 = u.Unit("m")
@@ -246,3 +246,27 @@ def test_convert_string():
 
     v2 = u1.to(unit_out="mm", value=20)
     assert v2 == 20000
+
+def test_copy():
+    u1=u.Unit("10 m")
+    u2=copy.copy(u1)
+    assert u1==u2
+    
+def test_deepcopy():
+    u1=u.Unit("10 m")
+    u2=copy.deepcopy(u1)
+    assert u1==u2
+    
+def test_deepcopy_object():
+    class val1:
+        value1=u.Unit("10 m")
+        value2=u.Unit("20 m")
+        value3=u.Unit("30 m")
+        values=[value1,value2,value3]
+        
+    v=val1()
+    v2=copy.deepcopy(v)
+    assert v2.value1==v.value1 
+    v.value1=u.Unit("50 m")
+    assert v2.value1==u.Unit("10 m")
+