From 0ff7331dbe87a52e4ab7cabb8f596bd3cf2788a8 Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Sat, 6 Apr 2019 14:00:52 -0700
Subject: [PATCH] Move axis_label method higher to general CoordinateSystems

---
 manimlib/mobject/coordinate_systems.py | 56 +++++++++++++-------------
 1 file changed, 28 insertions(+), 28 deletions(-)

diff --git a/manimlib/mobject/coordinate_systems.py b/manimlib/mobject/coordinate_systems.py
index ebce02f6..75424364 100644
--- a/manimlib/mobject/coordinate_systems.py
+++ b/manimlib/mobject/coordinate_systems.py
@@ -49,6 +49,34 @@ class CoordinateSystem():
     def get_z_axis(self):
         return self.get_axis(2)
 
+    def get_x_axis_label(self, label_tex, edge=RIGHT, direction=DL, **kwargs):
+        return self.get_axis_label(
+            label_tex, self.get_x_axis(),
+            edge, direction, **kwargs
+        )
+
+    def get_y_axis_label(self, label_tex, edge=UP, direction=DR, **kwargs):
+        return self.get_axis_label(
+            label_tex, self.get_y_axis(),
+            edge, direction, **kwargs
+        )
+
+    def get_axis_label(self, label_tex, axis, edge, direction, buff=MED_SMALL_BUFF):
+        label = TexMobject(label_tex)
+        label.next_to(
+            axis.get_edge_center(edge), direction,
+            buff=buff
+        )
+        label.shift_onto_screen(buff=MED_SMALL_BUFF)
+        return label
+
+    def get_axis_labels(self, x_label_tex="x", y_label_tex="y"):
+        self.axis_labels = VGroup(
+            self.get_x_axis_label(x_label_tex),
+            self.get_y_axis_label(y_label_tex),
+        )
+        return self.axis_labels
+
     def get_graph(self, function, **kwargs):
         x_min = kwargs.pop("x_min", self.x_min)
         x_max = kwargs.pop("x_max", self.x_max)
@@ -318,34 +346,6 @@ class NumberPlane(Axes):
     def get_axes(self):
         return self.axes
 
-    def get_x_axis_label(self, label_tex, edge=RIGHT, direction=DL, **kwargs):
-        return self.get_axis_label(
-            label_tex, self.get_x_axis(),
-            edge, direction, **kwargs
-        )
-
-    def get_y_axis_label(self, label_tex, edge=UP, direction=DR, **kwargs):
-        return self.get_axis_label(
-            label_tex, self.get_y_axis(),
-            edge, direction, **kwargs
-        )
-
-    def get_axis_label(self, label_tex, axis, edge, direction, buff=MED_SMALL_BUFF):
-        label = TexMobject(label_tex)
-        label.next_to(
-            axis.get_edge_center(edge), direction,
-            buff=buff
-        )
-        label.shift_onto_screen(buff=MED_SMALL_BUFF)
-        return label
-
-    def get_axis_labels(self, x_label_tex="x", y_label_tex="y"):
-        self.axis_labels = VGroup(
-            self.get_x_axis_label(x_label_tex),
-            self.get_y_axis_label(y_label_tex),
-        )
-        return self.axis_labels
-
     def get_vector(self, coords, **kwargs):
         kwargs["buff"] = 0
         return Arrow(