aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc')
-rw-r--r--tensorflow/contrib/tensorboard/db/summary_db_writer_test.cc56
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