fix(properties): use setter for columns property

Author: open-dynaMIX

anonip.py

@@ -89,9 +89,7 @@ def __init__(
         :param replace: str
         :param skip_private: bool
         """
-        columns = columns if columns else [1]
-        # change columns to be 0-based
-        self.columns = [c - 1 for c in columns]
+        self.columns = columns
         self._prefixes = {}  # next two lines will fill the values
         self.ipv4mask = ipv4mask
         self.ipv6mask = ipv6mask
@@ -100,6 +98,15 @@ def __init__(
         self.replace = replace
         self.skip_private = skip_private
 
+    @property
+    def columns(self):
+        return self._columns
+
+    @columns.setter
+    def columns(self, columns):
+        # change columns to be 0-based
+        self._columns = [c - 1 for c in columns] if columns else [0]
+
     @property
     def ipv4mask(self):
         return self._ipv4mask

tests.py

@@ -310,3 +310,10 @@ def test_properties_v6():
     a = anonip.Anonip(ipv4mask=11, ipv6mask=83)
     assert a.ipv6mask == 83
     assert a._prefixes[6] == 45
+
+
+def test_properties_columns():
+    a = anonip.Anonip()
+    assert a.columns == [0]
+    a.columns = [5, 6]
+    assert a.columns == [4, 5]