Union_Test.java

package jetbrains.mps.baseLanguage.collections.unittest.simple_operations;

/*Generated by MPS */

import junit.framework.TestCase;
import jetbrains.mps.internal.collections.runtime.Sequence;
import jetbrains.mps.internal.collections.runtime.ISequenceClosure;
import java.util.Iterator;
import jetbrains.mps.baseLanguage.closures.runtime.YieldingIterator;
import junit.framework.Assert;

public class Union_Test extends TestCase {
  public void test_union1() throws Exception {
    Iterable<Integer> s1 = Sequence.fromClosure(new ISequenceClosure<Integer>() {
      public Iterable<Integer> iterable() {
        return new Iterable<Integer>() {
          public Iterator<Integer> iterator() {
            return new YieldingIterator<Integer>() {
              private int __CP__ = 0;
              protected boolean moveToNext() {
__loop__:
                do {
__switch__:
                  switch (this.__CP__) {
                    case -1:
                      assert false : "Internal error";
                      return false;
                    case 2:
                      this.__CP__ = 3;
                      this.yield((Integer) 1);
                      return true;
                    case 3:
                      this.__CP__ = 4;
                      this.yield((Integer) 2);
                      return true;
                    case 4:
                      this.__CP__ = 1;
                      this.yield((Integer) 3);
                      return true;
                    case 0:
                      this.__CP__ = 2;
                      break;
                    default:
                      break __loop__;
                  }
                } while (true);
                return false;
              }
            };
          }
        };
      }
    });
    Iterable<Integer> s2 = Sequence.fromClosure(new ISequenceClosure<Integer>() {
      public Iterable<Integer> iterable() {
        return new Iterable<Integer>() {
          public Iterator<Integer> iterator() {
            return new YieldingIterator<Integer>() {
              private int __CP__ = 0;
              protected boolean moveToNext() {
__loop__:
                do {
__switch__:
                  switch (this.__CP__) {
                    case -1:
                      assert false : "Internal error";
                      return false;
                    case 2:
                      this.__CP__ = 3;
                      this.yield((Integer) 2);
                      return true;
                    case 3:
                      this.__CP__ = 4;
                      this.yield((Integer) 3);
                      return true;
                    case 4:
                      this.__CP__ = 1;
                      this.yield((Integer) 4);
                      return true;
                    case 0:
                      this.__CP__ = 2;
                      break;
                    default:
                      break __loop__;
                  }
                } while (true);
                return false;
              }
            };
          }
        };
      }
    });
    Iterable<Integer> s3 = Sequence.fromIterable(s1).union(Sequence.fromIterable(s2));
    Assert.assertSame(Sequence.fromIterable(s3).count(), 4);
    // order in s3 is undefined 
    Assert.assertTrue(Sequence.fromIterable(s3).contains(1));
    Assert.assertTrue(Sequence.fromIterable(s3).contains(2));
    Assert.assertTrue(Sequence.fromIterable(s3).contains(3));
    Assert.assertTrue(Sequence.fromIterable(s3).contains(4));
    // null test 
    s1 = null;
    Iterable<Integer> s2_ = Sequence.fromIterable(s1).union(Sequence.fromIterable(s2));
    Assert.assertEquals(3, Sequence.fromIterable(s2_).count());
    Iterable<Integer> s2__ = Sequence.fromIterable(s2).union(null);
    Assert.assertEquals(3, Sequence.fromIterable(s2__).count());
    Iterable<Integer> s4 = Sequence.fromIterable(s1).union(null);
    Assert.assertNull(Sequence.fromIterable(s4).first());
  }
}