TransformTest.java :  » Development » mango » uk » co » jezuk » mango » Java Open Source

package uk.co.jezuk.mango;

import junit.framework.*;
import java.util.List;
import java.util.ArrayList;

public class TransformTest extends TestCase
{
  public TransformTest(String name)
  {
    super(name);
  } // TransformTest

  public static Test suite()
  {
    return new TestSuite(TransformTest.class);
  } // suite

  public void test1()
  {
    ArrayList in = new ArrayList();
    in.add(new Integer(1));
    in.add(new Integer(2));
    in.add(new Integer(3));
    List out = (List)Algorithms.transform(in.iterator(), new Square(), new ArrayList());
    assertEquals(3, out.size());
    assertEquals(1, ((Integer)out.get(0)).intValue());
    assertEquals(4, ((Integer)out.get(1)).intValue());
    assertEquals(9, ((Integer)out.get(2)).intValue());
  } // test1
    
  public void test2()
  {
    ArrayList in = new ArrayList();
    in.add("A");
    in.add("B");
    in.add("C");
    List out = (List)Algorithms.transform(in.iterator(), new Duplicator(), new ArrayList());
    assertEquals(6, out.size());
    assertEquals("A", out.get(0));
    assertEquals("A", out.get(1));
    assertEquals("B", out.get(2));
    assertEquals("B", out.get(3));
    assertEquals("C", out.get(4));
    assertEquals("C", out.get(5));
  } // test2

  public class Square implements UnaryFunction
  {
    public Object fn(Object x)
    {
      if(x instanceof Integer)
  return new Integer(((Integer)x).intValue() * ((Integer)x).intValue());
      return null; 
    } // fn
  } // Square

  public class Duplicator implements UnaryFunction
  {
    public Object fn(Object x)
    {
      if(x instanceof String)
      {
        ArrayList ret = new ArrayList();
         ret.add(new String((String)x));
         ret.add(new String((String)x));
         return ret;
      } // if ...
      return null;
    } // fn
  } // Duplicator
} // TransformTest