small fix to type hints

Author: rsheftel

docs/change_log.rst

@@ -184,8 +184,12 @@ an installation requirement.
 - Remove travis-CI configs as it is no longer used
 - Merge coveragerc file into pyproject.toml
 
-3.2.0 (04/14/25)
+3.2.0 (04/15/25)
 ~~~~~~~~~~~~~~~~
 - Add type hints
 - Add as_namedtuple option to DataFrame get_columns() and get_location()
 - minimum python version now 3.11
+
+3.2.1 (04/15/25)
+~~~~~~~~~~~~~~~~
+- Fix to type hints

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "raccoon"
-version = "3.2.0"
+version = "3.2.1"
 authors = [
     { name = "Ryan Sheftel", email = "rsheftel@alumni.upenn.edu" },
 ]

raccoon/dataframe.py

@@ -29,7 +29,7 @@ class DataFrame(object):
 
     def __init__(
         self,
-        data: dict[Any, list] | None = None,
+        data: dict[Any, list | Any] | None = None,
         columns: list | None = None,
         index: list | None = None,
         index_name: str | tuple | None = "index",

tests/test_dataframe/test_dataframe.py

@@ -43,6 +43,19 @@ def check_list():
     check_list()
 
 
+def test_construction():
+    actual = rc.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}, index=["a", "b", "c"], columns=["b", "a"])
+    assert actual.data == [[4, 5, 6], [1, 2, 3]]
+    actual = rc.DataFrame({"a": 9, "b": 10})
+    assert actual.data == [[9], [10]]
+    actual = rc.DataFrame({"a": 9, "b": {"another": 90}}, index=["odd"])
+    assert actual.data == [[9], [{"another": 90}]]
+
+    # data length is mismatch with index length
+    with pytest.raises(ValueError):
+        rc.DataFrame({"a": 9, "b": {"another": 90}}, index=["a", "b", "c"])
+
+
 def test_to_dict():
     df = rc.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]}, index=['a', 'b', 'c'], columns=['b', 'a'])