Merge pull request #18 from tomicapretto/support_data_xycoords

Allow users to do xycoords='data'

flexitext/flexitext.py

@@ -55,7 +55,7 @@ def plot(
             Vertical alignment for text within multiline texts. Can be one of `"top"`, `"bottom"`,
             `"left"`, `"right"`, `"center"`, or `"baseline"`. Defaults to `"baseline"`.
         xycoords: str
-            The coordinate system for `x` and `y`. Must be one of `'axes fraction'` or
+            The coordinate system for `x` and `y`. Must be one of `'data'`, `'axes fraction'`, or
             `'figure fraction'`.
         ax: matplotlib.axes.Axes
             Matplotlib `Axes`. The default value means the `Axes` is obtained with `plt.gca()`
@@ -67,14 +67,15 @@ def plot(
 
         if ax is None:
             ax = plt.gca()
-        if xycoords == "axes fraction":
+        if xycoords in ("axes fraction", "data"):
             parent = ax
         elif xycoords == "figure fraction":
             parent = ax.figure
             xycoords = ax.figure.transFigure
         else:
             raise ValueError(
-                f"'xycoords' must be one of 'axes fraction' or 'figure fraction', not {xycoords}"
+                "'xycoords' must be one of 'data', 'axes fraction' or 'figure fraction', "
+                f"not {xycoords}"
             )
 
         offsetbox = self._make_offset_box(ma, mva)

flexitext/tests/test_flexitext.py

@@ -18,3 +18,9 @@ def test_flexitext():
 
     with pytest.raises(ValueError):
         flexitext(0.5, 0.5, text, xycoords="other stuff")
+
+    # Test that xycoords="data" works
+    _, ax = plt.subplots()
+    ax.plot([10, 20], [50, 80])
+    text = "<color:blue, size:16>This is blue text</> and this is regular text"
+    flexitext(15, 60, text, ha="center", xycoords="data")