diff options
Diffstat (limited to 'tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc')
-rw-r--r-- | tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc | 56 |
1 files changed, 1 insertions, 55 deletions
diff --git a/tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc b/tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc index c1af51e7b7..d32904f97c 100644 --- a/tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc +++ b/tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc @@ -14,19 +14,14 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/contrib/tensorboard/db/summary_db_writer.h" -#include "tensorflow/core/framework/summary.pb.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/lib/db/sqlite.h" -#include "tensorflow/core/lib/strings/strcat.h" #include "tensorflow/core/platform/env.h" #include "tensorflow/core/platform/test.h" -#include "tensorflow/core/util/event.pb.h" namespace tensorflow { namespace { -const float kTolerance = 1e-5; - Tensor MakeScalarInt64(int64 x) { Tensor t(DT_INT64, TensorShape({})); t.scalar<int64>()() = x; @@ -46,7 +41,7 @@ class FakeClockEnv : public EnvWrapper { class SummaryDbWriterTest : public ::testing::Test { protected: - void SetUp() override { db_ = Sqlite::Open(":memory:").ValueOrDie(); } + void SetUp() override { db_ = Sqlite::Open("file::memory:").ValueOrDie(); } void TearDown() override { if (writer_ != nullptr) { @@ -163,54 +158,5 @@ TEST_F(SummaryDbWriterTest, TensorsWritten_RowsGetInitialized) { QueryString("SELECT tensor FROM Tensors WHERE step = 2").empty()); } -TEST_F(SummaryDbWriterTest, EmptyParentNames_NoParentsCreated) { - TF_ASSERT_OK(CreateSummaryDbWriter(db_, "", "", "", &env_, &writer_)); - TF_ASSERT_OK(writer_->WriteTensor(1, MakeScalarInt64(123LL), "taggy", - "this-is-metaaa")); - TF_ASSERT_OK(writer_->Flush()); - ASSERT_EQ(0LL, QueryInt("SELECT COUNT(*) FROM Users")); - ASSERT_EQ(0LL, QueryInt("SELECT COUNT(*) FROM Experiments")); - ASSERT_EQ(0LL, QueryInt("SELECT COUNT(*) FROM Runs")); - ASSERT_EQ(1LL, QueryInt("SELECT COUNT(*) FROM Tags")); - ASSERT_EQ(1LL, QueryInt("SELECT COUNT(*) FROM Tensors")); -} - -TEST_F(SummaryDbWriterTest, WriteEvent_Scalar) { - TF_ASSERT_OK(CreateSummaryDbWriter(db_, "", "", "", &env_, &writer_)); - std::unique_ptr<Event> e{new Event}; - e->set_step(7); - e->set_wall_time(123.456); - Summary::Value* s = e->mutable_summary()->add_value(); - s->set_tag("π"); - s->set_simple_value(3.14f); - s = e->mutable_summary()->add_value(); - s->set_tag("φ"); - s->set_simple_value(1.61f); - TF_ASSERT_OK(writer_->WriteEvent(std::move(e))); - TF_ASSERT_OK(writer_->Flush()); - ASSERT_EQ(2LL, QueryInt("SELECT COUNT(*) FROM Tags")); - ASSERT_EQ(2LL, QueryInt("SELECT COUNT(*) FROM Tensors")); - int64 tag1_id = QueryInt("SELECT tag_id FROM Tags WHERE tag_name = 'π'"); - int64 tag2_id = QueryInt("SELECT tag_id FROM Tags WHERE tag_name = 'φ'"); - EXPECT_GT(tag1_id, 0LL); - EXPECT_GT(tag2_id, 0LL); - EXPECT_EQ(123.456, QueryDouble(strings::StrCat( - "SELECT computed_time FROM Tensors WHERE tag_id = ", - tag1_id, " AND step = 7"))); - EXPECT_EQ(123.456, QueryDouble(strings::StrCat( - "SELECT computed_time FROM Tensors WHERE tag_id = ", - tag2_id, " AND step = 7"))); - EXPECT_NEAR(3.14, - QueryDouble(strings::StrCat( - "SELECT tensor FROM Tensors WHERE tag_id = ", tag1_id, - " AND step = 7")), - kTolerance); // Summary::simple_value is float - EXPECT_NEAR(1.61, - QueryDouble(strings::StrCat( - "SELECT tensor FROM Tensors WHERE tag_id = ", tag2_id, - " AND step = 7")), - kTolerance); -} - } // namespace } // namespace tensorflow |