add test

Author: nschloe

test/test_scatter_different_colors.py

@@ -0,0 +1,49 @@
+from helpers import assert_equality
+
+
+def plot():
+    import numpy as np
+    from matplotlib import pyplot as plt
+
+    fig = plt.figure()
+    np.random.seed(123)
+    n = 4
+    plt.scatter(
+        np.random.rand(n),
+        np.random.rand(n),
+        color=np.array(
+            [
+                [1.0, 0.6, 0.0],
+                [0.0, 1.0, 0.0],
+                [0.0, 0.0, 1.0],
+                [0.0, 1.0, 1.0],
+            ]
+        ),
+        edgecolors=np.array(
+            [
+                [0.0, 1.0, 0.0],
+                [0.0, 0.0, 1.0],
+                [0.0, 1.0, 1.0],
+                [1.0, 0.0, 0.0],
+            ]
+        ),
+    )
+    return fig
+
+
+def test():
+    assert_equality(plot, __file__[:-3] + "_reference.tex")
+
+
+if __name__ == "__main__":
+    # import helpers
+    # helpers.compare_mpl_tex(plot)
+    # helpers.print_tree(plot())
+    plot()
+    import matplotlib.pyplot as plt
+
+    # plt.show()
+    # plt.savefig('out.pgf')
+    import tikzplotlib
+
+    tikzplotlib.save("out.tex", standalone=True)

test/test_scatter_different_colors_reference.tex

@@ -0,0 +1,39 @@
+\begin{tikzpicture}
+
+\begin{axis}[
+tick align=outside,
+tick pos=left,
+x grid style={white!69.019608!black},
+xmin=0.20337057, xmax=0.71995007,
+xtick style={color=black},
+y grid style={white!69.019608!black},
+ymin=0.39522357, ymax=1.0086471,
+ytick style={color=black}
+]
+\addplot [
+mark=*,
+only marks,
+scatter,
+scatter/@post marker code/.code={%
+  \endscope
+},
+scatter/@pre marker code/.code={%
+  \expanded{%
+  \noexpand\definecolor{thispointdrawcolor}{RGB}{\drawcolor}%
+  \noexpand\definecolor{thispointfillcolor}{RGB}{\fillcolor}%
+  }%
+  \scope[draw=thispointdrawcolor, fill=thispointfillcolor]%
+},
+visualization depends on={value \thisrow{draw} \as \drawcolor},
+visualization depends on={value \thisrow{fill} \as \fillcolor}
+]
+table{%
+x  y  draw  fill
+0.69646919 0.71946897 0,255,0 255,153,0
+0.28613933 0.42310646 0,0,255 0,255,0
+0.22685145 0.9807642 0,255,255 0,0,255
+0.55131477 0.68482974 255,0,0 0,255,255
+};
+\end{axis}
+
+\end{tikzpicture}