pydiverse-common 0.3.10__tar.gz → 0.3.11__tar.gz

{pydiverse_common-0.3.10 → pydiverse_common-0.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydiverse-common
-Version: 0.3.10
+Version: 0.3.11
 Summary: Common functionality shared between pydiverse libraries
 Author: QuantCo, Inc.
 Author-email: Martin Trautmann <windiana@users.sf.net>, Finn Rudolph <finn.rudolph@t-online.de>

{pydiverse_common-0.3.10 → pydiverse_common-0.3.11}/docs/source/changelog.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## 0.3.11 (2025-08-21)
+- fixed __eq__, __hash__, and __repr__ for types with parameters
+- import string length from sqlalchemy VARCHAR(n) type
+
 ## 0.3.10 (2025-08-21)
 - implemented String with max_length argument for SQL VARCHAR(n) generation
 - implemented Decimal with precision and scale arguments

{pydiverse_common-0.3.10 → pydiverse_common-0.3.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pydiverse-common"
-version = "0.3.10"
+version = "0.3.11"
 description = "Common functionality shared between pydiverse libraries"
 authors = [
   { name = "QuantCo, Inc." },

{pydiverse_common-0.3.10 → pydiverse_common-0.3.11}/src/pydiverse/common/dtypes.py

@@ -23,6 +23,10 @@ class Dtype:
         """Return a string representation of this dtype."""
         return self.__class__.__name__
 
+    def __str__(self):
+        """Return a string representation of this dtype."""
+        return self.__repr__()
+
     @classmethod
     def is_int(cls):
         """Return ``True`` if this dtype is an integer type."""
@@ -71,7 +75,7 @@ class Dtype:
             # otherwise.
             return Float64()
         if isinstance(sql_type, sa.String):
-            return String()
+            return String(sql_type.length)
         if isinstance(sql_type, sa.Boolean):
             return Bool()
         if isinstance(sql_type, sa.Date):
@@ -416,6 +420,20 @@ class Decimal(Float):
         self.precision = precision or 31
         self.scale = scale or (self.precision // 3 + 1)
 
+    def __eq__(self, rhs):
+        return (
+            isinstance(rhs, self.__class__)
+            and self.precision == rhs.precision
+            and self.scale == rhs.scale
+        )
+
+    def __hash__(self):
+        return hash((self.__class__.__name__, self.precision, self.scale))
+
+    def __repr__(self):
+        """Return a string representation of this dtype."""
+        return f"{self.__class__.__name__}({self.precision}, {self.scale})"
+
     def to_sql(self):
         import sqlalchemy as sa
 
@@ -480,6 +498,16 @@ class String(Dtype):
         """
         self.max_length = max_length
 
+    def __eq__(self, rhs):
+        return isinstance(rhs, self.__class__) and self.max_length == rhs.max_length
+
+    def __hash__(self):
+        return hash((self.__class__.__name__, self.max_length))
+
+    def __repr__(self):
+        """Return a string representation of this dtype."""
+        return f"{self.__class__.__name__}({self.max_length})"
+
     def to_sql(self):
         """Convert this Dtype to a SQL type."""
         import sqlalchemy as sa
@@ -523,10 +551,11 @@ class List(Dtype):
         return isinstance(rhs, List) and self.inner == rhs.inner
 
     def __hash__(self):
-        return hash((0, hash(self.inner)))
+        return hash((self.__class__.__name__, hash(self.inner)))
 
     def __repr__(self):
-        return f"List[{repr(self.inner)}]"
+        """Return a string representation of this dtype."""
+        return f"{self.__class__.__name__}[{self.inner}]"
 
     def to_sql(self):
         import sqlalchemy as sa
@@ -556,11 +585,13 @@ class Enum(String):
     def __eq__(self, rhs):
         return isinstance(rhs, Enum) and self.categories == rhs.categories
 
-    def __repr__(self) -> str:
-        return f"Enum[{', '.join(repr(c) for c in self.categories)}]"
-
     def __hash__(self):
-        return hash(tuple(self.categories))
+        return hash((self.__class__.__name__, tuple(self.categories)))
+
+    def __repr__(self) -> str:
+        return (
+            f"{self.__class__.__name__}[{', '.join(repr(c) for c in self.categories)}]"
+        )
 
     def to_polars(self):
         import polars as pl

{pydiverse_common-0.3.10 → pydiverse_common-0.3.11}/tests/dtypes/test_dtype_sqlalchemy.py

@@ -111,7 +111,7 @@ def test_all_types(type_):
     if type_ is pdc.List:
         type_obj = type_(pdc.Int64())
     elif type_ is pdc.Enum:
-        type_obj = type_("a", "b", "c")
+        type_obj = type_("a", "bbb", "cc")
     else:
         type_obj = type_()
     dst_type = type_obj.to_sql()
@@ -129,6 +129,6 @@ def test_all_types(type_):
         Float: Float64(),
         Int: Int64(),
         # there is no Enum
-        Enum: String(),
+        Enum: String(3),
     }
     assert back_type == acceptance_map.get(type_, type_obj)