Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
VisualDL
提交
3dcf637b
V
VisualDL
项目概览
PaddlePaddle
/
VisualDL
1 年多 前同步成功
通知
88
Star
4655
Fork
642
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
10
列表
看板
标记
里程碑
合并请求
2
Wiki
5
Wiki
分析
仓库
DevOps
项目成员
Pages
V
VisualDL
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
10
Issue
10
列表
看板
标记
里程碑
合并请求
2
合并请求
2
Pages
分析
分析
仓库分析
DevOps
Wiki
5
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
3dcf637b
编写于
12月 16, 2017
作者:
S
Superjom
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add sdk test
上级
2a29fa06
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
78 addition
and
0 deletion
+78
-0
visualdl/logic/sdk_test.cc
visualdl/logic/sdk_test.cc
+78
-0
未找到文件。
visualdl/logic/sdk_test.cc
0 → 100644
浏览文件 @
3dcf637b
#include "visualdl/logic/sdk.h"
#include <gtest/gtest.h>
namespace
visualdl
{
struct
ScalarTestHelper
{
ImHelper
rim
;
ImHelper
wim
;
const
std
::
string
dir
=
"./tmp/1.test"
;
void
operator
()(
std
::
function
<
void
()
>
read
,
std
::
function
<
void
()
>
write
)
{
wim
.
StartWriteSerice
(
dir
,
200
);
write
();
rim
.
StartReadService
(
dir
,
200
);
std
::
this_thread
::
sleep_for
(
std
::
chrono
::
milliseconds
(
600
));
read
();
}
};
TEST
(
Scalar
,
set_caption
)
{
ScalarTestHelper
helper
;
const
std
::
vector
<
std
::
string
>
captions
({
"train"
,
"test"
});
auto
write
=
[
&
]
{
auto
tablet
=
helper
.
wim
.
AddTablet
(
"tag0"
,
-
1
);
components
::
ScalarHelper
<
float
>
scalar
(
tablet
,
&
helper
.
wim
.
handler
());
scalar
.
SetCaptions
(
captions
);
};
auto
read
=
[
&
]
{
auto
mytablet
=
helper
.
rim
.
tablet
(
"tag0"
);
components
::
ScalarHelper
<
float
>
myscalar
(
mytablet
,
&
helper
.
rim
.
handler
());
auto
mycaptions
=
myscalar
.
GetCaptions
();
ASSERT_EQ
(
captions
,
mycaptions
);
};
helper
(
read
,
write
);
}
TEST
(
Scalar
,
add_records
)
{
ScalarTestHelper
helper
;
const
std
::
vector
<
std
::
string
>
captions
({
"train"
,
"test"
});
const
size_t
nsteps
=
100
;
auto
write
=
[
&
]
{
auto
tablet
=
helper
.
wim
.
AddTablet
(
"tag0"
,
-
1
);
components
::
ScalarHelper
<
float
>
scalar
(
tablet
,
&
helper
.
wim
.
handler
());
scalar
.
SetCaptions
(
captions
);
for
(
int
i
=
0
;
i
<
nsteps
;
i
++
)
{
scalar
.
AddRecord
(
i
*
10
,
std
::
vector
<
float
>
({(
float
)
i
,
(
float
)
i
+
1
}));
}
};
auto
read
=
[
&
]
{
auto
mytablet
=
helper
.
rim
.
tablet
(
"tag0"
);
components
::
ScalarHelper
<
float
>
myscalar
(
mytablet
,
&
helper
.
rim
.
handler
());
auto
records
=
myscalar
.
GetRecords
();
ASSERT_EQ
(
records
.
size
(),
nsteps
);
for
(
int
i
=
0
;
i
<
nsteps
;
i
++
)
{
ASSERT_EQ
(
records
[
i
],
std
::
vector
<
float
>
({(
float
)
i
,
(
float
)
i
+
1
}));
}
};
helper
(
read
,
write
);
}
}
// namespace visualdl
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录