astropy · elmostafa-abdelaal · Apr 5, 2025 · Apr 5, 2025 · Apr 5, 2025 · Apr 5, 2025
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -13,6 +13,8 @@ astropy.table
   default behavior will be to use the remote versions of jQuery and DataTables
   from a CDN. [#17480]
 
+
+
 Bug Fixes
 ---------
 
@@ -92,8 +94,8 @@ astropy.units
   ``.dtype``, independent of whether that is a numpy data type. [#17469]
 
 - The zebi (Zi, 2^70) and yobi (Yi, 2^80) binary prefixes are now supported. [#17692]
-
-astropy.visualization
+**Added**
+- Added `unit_support()` to enable both quantity and time support together. [#8860]
 ^^^^^^^^^^^^^^^^^^^^^
 
 - Fix ``CoordinateHelper.ticklabels``. The getter was incorrectly returning
@@ -245,7 +247,7 @@ astropy.units
 - Added support for calling numpy array constructors (``np.require``,
   ``np.identity``, ``np.eye``, ``np.tri``, ``np.genfromtxt`` and ``np.loadtxt``)
   with ``like=Quantity(...))`` . [#17130]
-
+Added `unit_support()` to enable both quantity and time support together. [#8860]
 astropy.utils
 ^^^^^^^^^^^^^
 
@@ -18353,6 +18355,8 @@ astropy.wcs
 
 - Minor VOTable fixes [#596]
 
+
+
 - Fixed how ``setup.py`` uses ``distribute_setup.py`` to prevent possible
   ``VersionConflict`` errors when an older version of distribute is already
   installed on the user's system. [#616, #640]

@@ -4,7 +4,7 @@
 
 import numpy as np
 
-__all__ = ["quantity_support"]
+__all__ = ["quantity_support", "unit_support"]  # ✅ أضفنا unit_support هنا
-__all__ = ["quantity_support", "unit_support"]  # ✅ أضفنا unit_support هنا
+__all__ = [
+    "quantity_support",
+    "unit_support",
+]
-__all__ = ["quantity_support", "unit_support"]  # ✅ أضفنا unit_support هنا
+__all__ = [
+    "quantity_support",
+    "unit_support",
+]
 __doctest_skip__ = ["quantity_support"]
 
 
@@ -21,17 +21,16 @@
       >>> with visualization.quantity_support():
       ...     plt.figure()
       ...     plt.plot([1, 2, 3] * u.m)
-      [...]
+      ...
       ...     plt.plot([101, 125, 150] * u.cm)
-      [...]
+      ...
       ...     plt.draw()
 
     Parameters
     ----------
     format : `astropy.units.format.Base` subclass or str
         The name of a format or a formatter class.  If not
         provided, defaults to ``latex_inline``.
-
     """
     from matplotlib import ticker, units
 
@@ -100,3 +99,33 @@
                 units.registry[u.Quantity] = self._original_converter[u.Quantity]
 
     return MplQuantityConverter()
+
+
+# Add the unit_support function to the module's __all__ list
+
+
+def unit_support(format="latex_inline"):
+    """
+    Enable both quantity and time support for matplotlib.
+
+    This is a convenience function that enables both astropy's
-    This is a convenience function that enables both astropy's
+    This is a convenience function that enables both
-    This is a convenience function that enables both astropy's
+    This is a convenience function that enables both
+    `quantity_support` and `time_support`.
+
+    Parameters
+    ----------
+    format : str
+        Format for unit labels (default: 'latex_inline').
+
+    Returns
+    -------
+    context : contextlib.ExitStack
+        A context manager that enables both supports.
+    """
+    from contextlib import ExitStack
+
+    from astropy.visualization import quantity_support, time_support
+
+    stack = ExitStack()
+    stack.enter_context(quantity_support(format=format))
+    stack.enter_context(time_support())
+    return stack
diff --git a/docs/changelog.rst b/docs/changelog.rst
@@ -8,3 +8,4 @@ Full Changelog
    :towncrier: ../
    :towncrier-skip-if-empty:
    :changelog_file: ../CHANGES.rst
+- Added `unit_support()` to enable both quantity and time support together. [#8860]
@@ -48,3 +48,10 @@ Reference/API
    :maxdepth: 2
 
    ref_api
+
+.. function:: unit_support(format="latex_inline")
+
+   Enable both quantity and time support for matplotlib.
+
+   :param format: Format for unit labels (default: 'latex_inline').
+   :returns: Context manager enabling both supports.