From b84ffab5b6ec8f205cd0155191b9ffaa03213006 Mon Sep 17 00:00:00 2001
From: Shiwei Tong <tongsw@mail.ustc.edu.cn>
Date: Sat, 22 May 2021 21:25:18 +0800
Subject: [PATCH] [example] update

---
 examples/formula/formula.py | 41 +++++++++++++++++++++++++++++--------
 1 file changed, 32 insertions(+), 9 deletions(-)

diff --git a/examples/formula/formula.py b/examples/formula/formula.py
index ec6dd4f9..09585f19 100644
--- a/examples/formula/formula.py
+++ b/examples/formula/formula.py
@@ -1,13 +1,36 @@
 # coding: utf-8
 # 2021/3/8 @ tongshiwei
+#
+from EduNLP.Formula import Formula, FormulaGroup, link_formulas
+#
+# f1 = Formula(r"x + y", variable_standardization=True)
+# f2 = Formula(r"y + x", variable_standardization=True)
+# f3 = Formula(r"z + y", variable_standardization=True)
+#
+# print(f1.element)
+# print(f2.element)
+# print(f3.element)
+#
+# print("-----------------------")
+#
+# link_formulas(f1, f2, f3)
+#
+# print("------------------------")
+#
+# print(f1.element)
+# print(f2.element)
+# print(f3.element)
+#
+# print("---------------------")
+#
+# fg = FormulaGroup(
+#     [r"x + y", r"y + x", r"y + z"]
+# )
+# for f in fg:
+#     print(f.element)
 
-import matplotlib.pyplot as plt
-from EduNLP.Formula import Formula
-from EduNLP.Formula.viz import ForestPlotter
+# fg = FormulaGroup(["x", "y", "x"])
+# print(fg.elements)
 
-f = Formula(r"\frac{\sqrt{x^2}}{\pi} + 1 = y", variable_standardization=True)
-
-ForestPlotter().export(
-    f.ast, root_list=[node["val"]["id"] for node in f.element if node["structure"]["father"] is None],
-)
-plt.show()
\ No newline at end of file
+fg = FormulaGroup(["x", Formula("y"), "x"])
+print(fg.elements)
\ No newline at end of file