/* Copyright (C) 2006 Univ. of Massachusetts Amherst, Computer Science Dept.
   This file is part of "MALLET" (MAchine Learning for LanguagE Toolkit).
   http://www.cs.umass.edu/~mccallum/mallet
   This software is provided under the terms of the Common Public License,
   version 1.0, as published by http://www.opensource.org.  For further
   information, see the file `LICENSE' included with this distribution. */

package cc.mallet.grmm.test;

import junit.framework.*;

import java.util.Iterator;
import java.util.Random;

import cc.mallet.grmm.inference.RandomGraphs;
import cc.mallet.grmm.inference.TRP;
import cc.mallet.grmm.types.Factor;
import cc.mallet.grmm.types.FactorGraph;

/**
 * $Id: TestTRP.java,v 1.1 2007/10/22 21:37:40 mccallum Exp $
 */
public class TestTRP extends TestCase {

  public TestTRP (String name)
  {
    super (name);
  }

  public void testEarlyStopping ()
  {
    FactorGraph grid = RandomGraphs.randomAttractiveGrid (5, 0.5, new Random (2413421));
    TRP trp = new TRP (new TRP.IterationTerminator (1));
    trp.setRandomSeed (14312341);

    trp.computeMarginals (grid);

    boolean oneIsDifferent = false;

    // check no exceptions thrown when asking for all marginals,
    //  and check that at least one factors' belief has changed
    //  from the choice at zero iterations.
    for (Iterator it = grid.factorsIterator (); it.hasNext();) {
      Factor f = (Factor) it.next ();
      Factor marg = trp.lookupMarginal (f.varSet ());// test no exception thrown
      if (!marg.almostEquals (f.duplicate ().normalize ())) {
        oneIsDifferent = true;
      }
    }

    assertTrue (oneIsDifferent);
  }

  /**
   * @return a <code>TestSuite</code>
   */
  public static TestSuite suite ()
  {
    return new TestSuite (TestTRP.class);
  }

  public static void main (String[] args)
  {
    TestSuite theSuite;
    if (args.length > 0) {
      theSuite = new TestSuite ();
      for (int i = 0; i < args.length; i++) {
        theSuite.addTest (new TestTRP (args[i]));
      }
    } else {
      theSuite = suite ();
    }

    junit.textui.TestRunner.run (theSuite);
  }

}