Domain: Add copy method

biolab · Nov 3, 2017 · fa91921 · fa91921
1 parent 9225e8d
commit fa91921
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Orange/data/domain.py b/Orange/data/domain.py
@@ -444,6 +444,14 @@ def _compute_col_indices(self, col_idx):
     def checksum(self):
         return hash(self)
 
+    def copy(self):
+        return Domain(
+            attributes=[a.make_proxy() for a in self.attributes],
+            class_vars=[a.make_proxy() for a in self.class_vars],
+            metas=[a.make_proxy() for a in self.metas],
+            source=self,
+        )
+
     def __eq__(self, other):
         if not isinstance(other, Domain):
             return False

diff --git a/Orange/tests/test_domain.py b/Orange/tests/test_domain.py
@@ -490,6 +490,17 @@ def test_domain_conversion_is_fast_enough(self):
 
         self.assertLessEqual(time() - start, 1)
 
+    def test_copy(self):
+        attributes = (age, gender, income)
+        new_attributes = (age, )
+
+        domain = Domain(attributes, [race], [ssn])
+
+        new_domain = domain.copy()
+        new_domain.attributes = new_attributes
+
+        self.assertEqual(domain.attributes, attributes)
+
 
 class TestDomainFilter(unittest.TestCase):
     def setUp(self):