add hash method

Author: Tishj

src/duckdb_py/include/duckdb_python/pytype.hpp

@@ -30,6 +30,7 @@ class DuckDBPyType : public enable_shared_from_this<DuckDBPyType> {
 
 public:
 	bool Equals(const shared_ptr<DuckDBPyType> &other) const;
+	ssize_t Hash() const;
 	bool EqualsString(const string &type_str) const;
 	shared_ptr<DuckDBPyType> GetAttribute(const string &name) const;
 	py::list Children() const;

src/duckdb_py/typing/pytype.cpp

@@ -46,6 +46,10 @@ bool DuckDBPyType::Equals(const shared_ptr<DuckDBPyType> &other) const {
 	return type == other->type;
 }
 
+ssize_t DuckDBPyType::Hash() const {
+	return py::hash(py::str(ToString()));
+}
+
 bool DuckDBPyType::EqualsString(const string &type_str) const {
 	return StringUtil::CIEquals(type.ToString(), type_str);
 }
@@ -328,6 +332,7 @@ void DuckDBPyType::Initialize(py::handle &m) {
 	type_module.def("__repr__", &DuckDBPyType::ToString, "Stringified representation of the type object");
 	type_module.def("__eq__", &DuckDBPyType::Equals, "Compare two types for equality", py::arg("other"), py::is_operator());
 	type_module.def("__eq__", &DuckDBPyType::EqualsString, "Compare two types for equality", py::arg("other"), py::is_operator());
+	type_module.def("__hash__", &DuckDBPyType::Hash, "Hashes the type, equal to stringifying+hashing");
 	type_module.def_property_readonly("id", &DuckDBPyType::GetId);
 	type_module.def_property_readonly("children", &DuckDBPyType::Children);
 	type_module.def(py::init<>([](const string &type_str, shared_ptr<DuckDBPyConnection> connection = nullptr) {

tests/fast/test_type.py

@@ -214,6 +214,20 @@ def test_struct_from_dict(self):
         res = duckdb.list_type({'a': VARCHAR, 'b': VARCHAR})
         assert res == 'STRUCT(a VARCHAR, b VARCHAR)[]'
 
+    def test_hash_method(self):
+        type1 = duckdb.list_type({'a': VARCHAR, 'b': VARCHAR})
+        type2 = duckdb.list_type({'b': VARCHAR, 'a': VARCHAR})
+        type3 = VARCHAR
+
+        type_set = set()
+        type_set.add(type1)
+        type_set.add(type2)
+        type_set.add(type3)
+
+        type_set.add(type1)
+        expected = ['STRUCT(a VARCHAR, b VARCHAR)[]', 'STRUCT(b VARCHAR, a VARCHAR)[]', 'VARCHAR']
+        assert sorted([str(x) for x in list(type_set)]) == expected
+
     # NOTE: we can support this, but I don't think going through hoops for an outdated version of python is worth it
     @pytest.mark.skipif(sys.version_info < (3, 9), reason="python3.7 does not store Optional[..] in a recognized way")
     def test_optional(self):