match() (find()): use exmap (exset) to store matched subexpressions.

[ginac.git] / check / exam_color.cpp

diff --git a/check/exam_color.cpp b/check/exam_color.cpp
index cbc99462ef676765d8a2501da44830972871a2ab..53c72c010b8a2940b1361efd7c01b941243e32ec 100644 (file)
--- a/check/exam_color.cpp
+++ b/check/exam_color.cpp
@@ -3,7 +3,7 @@
  *  Here we test GiNaC's color objects (su(3) Lie algebra). */
 
 /*
- *  GiNaC Copyright (C) 1999-2003 Johannes Gutenberg University Mainz, Germany
+ *  GiNaC Copyright (C) 1999-2008 Johannes Gutenberg University Mainz, Germany
  *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
@@ -17,10 +17,13 @@
  *
  *  You should have received a copy of the GNU General Public License
  *  along with this program; if not, write to the Free Software
- *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  */
 
-#include "exams.h"
+#include <iostream>
+#include "ginac.h"
+using namespace std;
+using namespace GiNaC;
 
 static unsigned check_equal(const ex &e1, const ex &e2)
 {
@@ -113,7 +116,7 @@ static unsigned color_check3()
 
        unsigned result = 0;
 
-       idx a(symbol("a"), 8), b(symbol("b"), 8), c(symbol("c"), 8), k(symbol("k"), 8);
+       idx a(symbol("a"), 8), b(symbol("b"), 8), c(symbol("c"), 8);
        ex e;
 
        e = color_ONE();
@@ -125,6 +128,18 @@ static unsigned color_check3()
        e = color_T(a) * color_T(b) * color_T(c);
        result += check_equal(color_trace(e), color_h(a, b, c) / 4);
 
+       e = color_ONE(0) * color_ONE(1) / 9;
+       result += check_equal(color_trace(e, 0), color_ONE(1) / 3);
+       result += check_equal(color_trace(e, 1), color_ONE(0) / 3);
+       result += check_equal(color_trace(e, 2), e);
+       result += check_equal(color_trace(e, lst(0, 1)), 1);
+
+       e = color_T(a, 0) * color_T(a, 1) * color_T(b, 0) * color_T(b, 1);
+       result += check_equal_simplify(color_trace(e, 0), 2 * color_ONE(1) / 3);
+       result += check_equal_simplify(color_trace(e, 1), 2 * color_ONE(0) / 3);
+       result += check_equal_simplify(color_trace(e, 2), e);
+       result += check_equal_simplify(color_trace(e, lst(0, 1)), 2);
+
        return result;
 }
 
@@ -133,18 +148,15 @@ unsigned exam_color()
        unsigned result = 0;
        
        cout << "examining color objects" << flush;
-       clog << "----------color objects:" << endl;
 
        result += color_check1();  cout << '.' << flush;
        result += color_check2();  cout << '.' << flush;
        result += color_check3();  cout << '.' << flush;
        
-       if (!result) {
-               cout << " passed " << endl;
-               clog << "(no output)" << endl;
-       } else {
-               cout << " failed " << endl;
-       }
-       
        return result;
 }
+
+int main(int argc, char** argv)
+{
+       return exam_color();
+}