test/com/google/caliper/DefaultBenchmarkSuiteTest.java

/*
 * Copyright (C) 2009 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.caliper;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;

import junit.framework.TestCase;

import java.util.Arrays;
import java.util.Collection;

public class DefaultBenchmarkSuiteTest extends TestCase {

  public void testIntrospection() {
    SampleBenchmarkSuite suite = new SampleBenchmarkSuite();
    assertEquals(ImmutableSet.of("a", "b"), suite.parameterNames());
    assertEquals(ImmutableSet.of("1", "2", "3"), suite.parameterValues("a"));
    assertEquals(ImmutableSet.of("4"), suite.parameterValues("b"));
    assertEquals(ImmutableSet.of(
        SampleBenchmarkSuite.MultiplyBenchmark.class,
        SampleBenchmarkSuite.DivideBenchmark.class),
        suite.benchmarkClasses());

  }

  public void testCreateBenchmark() {
    SampleBenchmarkSuite originalSuite = new SampleBenchmarkSuite();
    Benchmark benchmark = originalSuite.createBenchmark(
        SampleBenchmarkSuite.MultiplyBenchmark.class,
        ImmutableMap.of("a", "2", "b", "4"));

    SampleBenchmarkSuite.MultiplyBenchmark multiplyBenchmark
        = (SampleBenchmarkSuite.MultiplyBenchmark) benchmark;

    SampleBenchmarkSuite multiplySuite = multiplyBenchmark.suite();
    assertNotSame(originalSuite, multiplySuite);
    assertEquals(2, multiplySuite.a);
    assertEquals(4, multiplySuite.b);
  }

  static class SampleBenchmarkSuite extends DefaultBenchmarkSuite {
    @Param int a;

    private static Collection<Integer> aValues = Arrays.asList(1, 2, 3);

    @Param int b;

    private static Collection<Integer> bValues() {
      return Arrays.asList(4);
    }

    class MultiplyBenchmark extends Benchmark {
      @Override public Object run(int trials) throws Exception {
        int result = 0;
        for (int i = 0; i < trials; i++) {
          result ^= a * b;
        }
        return result;
      }

      SampleBenchmarkSuite suite() {
        return SampleBenchmarkSuite.this;
      }
    }

    class DivideBenchmark extends Benchmark {
      @Override public Object run(int trials) throws Exception {
        int result = 0;
        for (int i = 0; i < trials; i++) {
          result ^= a / b;
        }
        return result;
      }
    }
  }
}