diff options
author | 2018-01-05 09:58:08 -0800 | |
---|---|---|
committer | 2018-01-05 10:02:07 -0800 | |
commit | 003ae0eab1d9a95a85e07887cdecc3e8c836b6e3 (patch) | |
tree | e649d6f2785bf83fe8ab51753cbc2f0254b3060c | |
parent | 2931079cb20f4e9682f4ad3529141e93125690e5 (diff) |
Added a couple of trivial tests
PiperOrigin-RevId: 180940645
-rw-r--r-- | tensorflow/core/grappler/clusters/single_machine_test.cc | 4 | ||||
-rw-r--r-- | tensorflow/core/grappler/clusters/virtual_cluster_test.cc | 4 |
2 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/clusters/single_machine_test.cc b/tensorflow/core/grappler/clusters/single_machine_test.cc index df936efad1..26841003c8 100644 --- a/tensorflow/core/grappler/clusters/single_machine_test.cc +++ b/tensorflow/core/grappler/clusters/single_machine_test.cc @@ -58,6 +58,10 @@ class SingleMachineTest : public ::testing::Test { std::unique_ptr<SingleMachine> cluster_; }; +TEST_F(SingleMachineTest, ClusterType) { + CHECK_EQ("single_machine", cluster_->type()); +} + TEST_F(SingleMachineTest, CostModel) { TrivialTestGraphInputYielder fake_input(4, 1, 10, false, cluster_->GetDeviceNames()); diff --git a/tensorflow/core/grappler/clusters/virtual_cluster_test.cc b/tensorflow/core/grappler/clusters/virtual_cluster_test.cc index fd925a6ce7..357b306b93 100644 --- a/tensorflow/core/grappler/clusters/virtual_cluster_test.cc +++ b/tensorflow/core/grappler/clusters/virtual_cluster_test.cc @@ -56,6 +56,10 @@ class VirtualClusterTest : public ::testing::Test { std::unique_ptr<VirtualCluster> cluster_; }; +TEST_F(VirtualClusterTest, ClusterType) { + CHECK_EQ("virtual", cluster_->type()); +} + TEST_F(VirtualClusterTest, CostModel) { TrivialTestGraphInputYielder fake_input(4, 1, 10, false, cluster_->GetDeviceNames()); |