| @@ -14,6 +14,7 @@ | |||||
| # ============================================================================ | # ============================================================================ | ||||
| """Summary cpu st.""" | """Summary cpu st.""" | ||||
| import os | import os | ||||
| import platform | |||||
| import tempfile | import tempfile | ||||
| import numpy as np | import numpy as np | ||||
| @@ -58,22 +59,23 @@ def train_summary_record(test_writer, steps): | |||||
| return out_me_dict | return out_me_dict | ||||
| class TestCpuSummary: | |||||
| """Test cpu summary.""" | |||||
| @pytest.mark.level0 | |||||
| @pytest.mark.platform_x86_cpu | |||||
| @pytest.mark.env_onecard | |||||
| def test_summary_step2_summary_record1(): | |||||
| """Test record 10 step summary.""" | |||||
| if platform.system() == "Windows": | |||||
| # Summary does not support windows currently. | |||||
| return | |||||
| @pytest.mark.level0 | |||||
| @pytest.mark.platform_x86_cpu_training | |||||
| @pytest.mark.env_onecard | |||||
| def test_summary_step2_summary_record1(self): | |||||
| """Test record 10 step summary.""" | |||||
| with tempfile.TemporaryDirectory() as tmp_dir: | |||||
| steps = 2 | |||||
| with SummaryRecord(tmp_dir) as test_writer: | |||||
| train_summary_record(test_writer, steps=steps) | |||||
| with tempfile.TemporaryDirectory() as tmp_dir: | |||||
| steps = 2 | |||||
| with SummaryRecord(tmp_dir) as test_writer: | |||||
| train_summary_record(test_writer, steps=steps) | |||||
| file_name = os.path.realpath(test_writer.full_file_name) | |||||
| with SummaryReader(file_name) as summary_writer: | |||||
| for _ in range(steps): | |||||
| event = summary_writer.read_event() | |||||
| tags = set(value.tag for value in event.summary.value) | |||||
| assert tags == {'tensor', 'histogram', 'scalar', 'image'} | |||||
| file_name = os.path.realpath(test_writer.full_file_name) | |||||
| with SummaryReader(file_name) as summary_writer: | |||||
| for _ in range(steps): | |||||
| event = summary_writer.read_event() | |||||
| tags = set(value.tag for value in event.summary.value) | |||||
| assert tags == {'tensor', 'histogram', 'scalar', 'image'} | |||||