/*
 * Copyright (C) 2008 The Guava Authors
 *
 * 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.common.collect;
public class TablesTest_gwt extends com.google.gwt.junit.client.GWTTestCase {
@Override public String getModuleName() {
  return "com.google.common.collect.testModule";
}
public void testEntryEquals() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testEntryEquals();
}

public void testEntryEqualsNull() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testEntryEqualsNull();
}

public void testImmutableEntryToString() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testImmutableEntryToString();
}

public void testToTable() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testToTable();
}

public void testToTableConflict() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testToTableConflict();
}

public void testToTableMerging() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testToTableMerging();
}

public void testToTableNullMerge() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testToTableNullMerge();
}

public void testToTableNullValues() throws Exception {
  com.google.common.collect.TablesTest testCase = new com.google.common.collect.TablesTest();
  testCase.testToTableNullValues();
}
}
