Skip to content

BUG: is_dtype_equal(dtype, "string[pyarrow]") raises if pyarrow not installed #44327

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 20 commits into from
Dec 15, 2021
Merged

BUG: is_dtype_equal(dtype, "string[pyarrow]") raises if pyarrow not installed #44327

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
542032c
import error resolved
shubham11941140 Nov 5, 2021
76ab6fa
pre-commit
shubham11941140 Nov 5, 2021
c82d836
Update test_misc.py
shubham11941140 Nov 5, 2021
c35e162
Changes done
shubham11941140 Nov 7, 2021
cff657a
Duplicate test removed
shubham11941140 Nov 7, 2021
dce9db6
Changes done
shubham11941140 Nov 8, 2021
5abfa2e
raises modified
shubham11941140 Nov 8, 2021
fb0bd2a
changes done
shubham11941140 Nov 23, 2021
f7b8888
decorator fix
shubham11941140 Nov 25, 2021
2de4936
Changed tests
shubham11941140 Nov 26, 2021
66327e7
precommit
shubham11941140 Nov 26, 2021
9566360
StringDtype added
shubham11941140 Nov 26, 2021
0f35541
Try-Except Changed
shubham11941140 Dec 8, 2021
7ea6b15
Changes Done
shubham11941140 Dec 8, 2021
9b0dc6c
Merge branch 'master' into b5
shubham11941140 Dec 14, 2021
b0a75fc
remove np.dtype from isinstance check
jorisvandenbossche Dec 14, 2021
75a4b26
remove np.dtype from isinstance check, catch ImportError instead
jorisvandenbossche Dec 14, 2021
f5bedc6
update whatsnew
jorisvandenbossche Dec 14, 2021
9c68f5c
fix rst syntax in whatsnew
jorisvandenbossche Dec 14, 2021
c550a00
clean-up
jorisvandenbossche Dec 14, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion doc/source/whatsnew/v1.3.5.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ Fixed regressions

Bug fixes
~~~~~~~~~
-
- Fixed bug in :bug:`is_dtype_equal` raises if pyarrow not installed (:issue:`44327`)
-

.. ---------------------------------------------------------------------------
Expand Down
5 changes: 4 additions & 1 deletion pandas/core/dtypes/common.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -599,9 +599,12 @@ def is_dtype_equal(source, target) -> bool:
if not isinstance(source, str):
# GH#38516 ensure we get the same behavior from
# is_dtype_equal(CDT, "category") and CDT == "category"
# GH-44276 ensures behaviour is consistent
# is_dtype_equal (dtype, "string[pyarrow]")
try:
src = get_dtype(source)
if isinstance(src, ExtensionDtype):
if isinstance(src, (ExtensionDtype, np.dtype)):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

is this change still needed?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It is needed because dtypes such as int64 float64 and other dtypes actually are np.dtype, so if I am not adding this check, the test is failing.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It is needed because dtypes such as int64 float64 and other dtypes actually are np.dtype, so if I am not adding this check, the test is failing.

It is not needed if we catch ImportError more generally, I think; I pushed some small commits to verify this.


return src == target
except (TypeError, AttributeError):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think you can probably catch the ImportError here (so it always returns False if the above raises an ImportError)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done

return False
Expand Down
6 changes: 6 additions & 0 deletions pandas/tests/dtypes/test_dtypes.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -115,6 +115,12 @@ def test_construction_from_string(self, dtype):
with pytest.raises(TypeError, match=msg):
CategoricalDtype.construct_from_string("foo")

def test_pyarrow_string_import_error(self, dtype):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Another note: this test is added to TestCategoricalDtype, so categorical specific tests. I think we should move it to a more generic location. For example in tests/dtypes/test_common.py there is a test_dtype_equal that tests the basic functionality of is_dtype_equal. Maybe add the new test there (or in the arrow string dtype specific tests, @jbrockmendel any preference?)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Request you to inform me where do I exactly put the test, in which file and which function

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You can put it in tests/dtypes/test_common.py, just after the existing test_dtype_equal test.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done

# GH-44276
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can you flesh this out a tiny bit e.g. 'dont raise if pyarrow is not installed'

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done

assert not is_dtype_equal(dtype, "string[pyarrow]")
with pytest.raises(ImportError, match="pyarrow"):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

umm we don't do this, rather use the decorator to skip the test

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done

yield

def test_constructor_invalid(self):
msg = "Parameter 'categories' must be list-like"
with pytest.raises(TypeError, match=msg):
Expand Down