fix code style of test_dataset.py

test=develop
revert-16555-model_data_cryption_link_all_lib
xjqbest 6 years ago committed by dongdaxiang
parent 45eb6f0765
commit 1073b4d8f9

@ -21,9 +21,9 @@ import unittest
class TestDataset(unittest.TestCase):
""" TestCases for Dataset """
""" TestCases for Dataset. """
def test_dataset_create(self):
""" Testcase for dataset create """
""" Testcase for dataset create. """
try:
dataset = fluid.DatasetFactory().create_dataset("InMemoryDataset")
except:
@ -41,7 +41,7 @@ class TestDataset(unittest.TestCase):
self.assertTrue(True)
def test_dataset_config(self):
""" Testcase for dataset configuration """
""" Testcase for dataset configuration. """
dataset = fluid.core.Dataset("MultiSlotDataset")
dataset.set_thread_num(12)
dataset.set_filelist(["a.txt", "b.txt", "c.txt"])
@ -66,7 +66,7 @@ class TestDataset(unittest.TestCase):
def test_in_memory_dataset_run(self):
"""
Testcase for InMemoryDataset from create to run
Testcase for InMemoryDataset from create to run.
"""
with open("test_in_memory_dataset_run_a.txt", "w") as f:
data = "1 1 2 3 3 4 5 5 5 5 1 1\n"
@ -112,7 +112,7 @@ class TestDataset(unittest.TestCase):
def test_queue_dataset_run(self):
"""
Testcase for QueueDataset from create to run
Testcase for QueueDataset from create to run.
"""
with open("test_queue_dataset_run_a.txt", "w") as f:
data = "1 1 2 3 3 4 5 5 5 5 1 1\n"

Loading…
Cancel
Save