Add the "axis equal" parameter

Author: Gorshanov Vadim

src/tikzplotlib/_axes.py

@@ -239,6 +239,8 @@ def _set_axis_dimensions(self, data, aspect_num, xlim, ylim):
         else:
             # TODO keep an eye on https://tex.stackexchange.com/q/480058/13262
             pass
+        if data["axis_equal"]:
+            self.axis_options.append("axis equal")
 
     def _ticks(self, data, obj):
         # get ticks

src/tikzplotlib/_save.py

@@ -41,6 +41,7 @@ def get_tikz_code(
     table_row_sep: str = "\n",
     flavor: str = "latex",
     every_n_dot: int = 1,
+    axis_equal: bool = False,
 ):
     """Main function. Here, the recursion into the image starts and the
     contents are picked up. The actual file gets written in this routine.
@@ -140,6 +141,9 @@ def get_tikz_code(
     :param every_n_dot: if path is encountered, only draw every Nth dot
     :type every_n_dot: int
 
+    :param axis_equal: if true, have equal axis ratio
+    :type axis_equal: bool
+
     :returns: None
 
     The following optional attributes of matplotlib's objects are recognized
@@ -181,6 +185,7 @@ def get_tikz_code(
     data["add axis environment"] = add_axis_environment
     data["show_info"] = show_info
     data["every n dot"] = every_n_dot
+    data["axis_equal"] = axis_equal
     # rectangle_legends is used to keep track of which rectangles have already
     # had \addlegendimage added. There should be only one \addlegenimage per
     # bar chart data series.