address a comment

Author: HyukjinKwon

python/pyarrow/src/arrow/python/inference.cc

@@ -339,6 +339,18 @@ class NumPyDtypeUnifier {
  private:
   static const char* DatetimeUnitName(NPY_DATETIMEUNIT unit) {
     switch (unit) {
+      case NPY_FR_Y:
+        return "Y";
+      case NPY_FR_M:
+        return "M";
+      case NPY_FR_W:
+        return "W";
+      case NPY_FR_D:
+        return "D";
+      case NPY_FR_h:
+        return "h";
+      case NPY_FR_m:
+        return "m";
       case NPY_FR_s:
         return "s";
       case NPY_FR_ms:
@@ -347,8 +359,12 @@ class NumPyDtypeUnifier {
         return "us";
       case NPY_FR_ns:
         return "ns";
-      case NPY_FR_D:
-        return "D";
+      case NPY_FR_ps:
+        return "ps";
+      case NPY_FR_fs:
+        return "fs";
+      case NPY_FR_as:
+        return "as";
       case NPY_FR_GENERIC:
         return "generic";
       default:

python/pyarrow/tests/test_array.py

@@ -2493,6 +2493,30 @@ def test_array_from_different_numpy_datetime_units_raises():
         pa.array(data)
 
 
+@pytest.mark.numpy
+@pytest.mark.parametrize('unit,unit_name', [
+    ('Y', 'Y'),  # year
+    ('M', 'M'),  # month
+    ('W', 'W'),  # week
+    ('h', 'h'),  # hour
+    ('m', 'm'),  # minute
+    ('ps', 'ps'),  # picosecond
+    ('fs', 'fs'),  # femtosecond
+    ('as', 'as'),  # attosecond
+])
+def test_array_from_unsupported_numpy_datetime_unit_names(unit, unit_name):
+    # Test that unsupported datetime units show meaningful names in error messages
+    s_data = [np.datetime64('2020-01-01', 's')]
+    unsupported_data = [np.datetime64('2020', unit)]
+
+    # Mix supported unit (s) with unsupported unit
+    data = s_data + unsupported_data
+
+    with pytest.raises(pa.ArrowInvalid,
+                       match=f"Cannot mix NumPy datetime64 units s and {unit_name}"):
+        pa.array(data)
+
+
 @pytest.mark.numpy
 @pytest.mark.parametrize('unit', ['ns', 'us', 'ms', 's'])
 def test_array_from_list_of_timestamps(unit):