Skip to content

Commit 7d2d687

Browse files
authored
cleanup: alphabetize argument order for FakeFlags (#5544)
* sort FakeFlags fields in core_plugin_test.py * data_ingester_test.py: clean up FakeFlags arg order * application_test.py: sort FakeFlags args, make logdir a kwarg
1 parent 7154c60 commit 7d2d687

File tree

3 files changed

+52
-42
lines changed

3 files changed

+52
-42
lines changed

tensorboard/backend/application_test.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -33,31 +33,31 @@
3333
class FakeFlags(object):
3434
def __init__(
3535
self,
36-
logdir,
36+
generic_data="auto",
37+
logdir="",
3738
logdir_spec="",
38-
purge_orphaned_data=True,
39-
reload_interval=60,
40-
samples_per_plugin=None,
4139
max_reload_threads=1,
42-
reload_task="auto",
43-
window_title="",
4440
path_prefix="",
41+
purge_orphaned_data=True,
42+
reload_interval=60,
4543
reload_multifile=False,
4644
reload_multifile_inactive_secs=4000,
47-
generic_data="auto",
45+
reload_task="auto",
46+
samples_per_plugin=None,
47+
window_title="",
4848
):
49+
self.generic_data = generic_data
4950
self.logdir = logdir
5051
self.logdir_spec = logdir_spec
51-
self.purge_orphaned_data = purge_orphaned_data
52-
self.reload_interval = reload_interval
53-
self.samples_per_plugin = samples_per_plugin or {}
5452
self.max_reload_threads = max_reload_threads
55-
self.reload_task = reload_task
56-
self.window_title = window_title
5753
self.path_prefix = path_prefix
54+
self.purge_orphaned_data = purge_orphaned_data
55+
self.reload_interval = reload_interval
5856
self.reload_multifile = reload_multifile
5957
self.reload_multifile_inactive_secs = reload_multifile_inactive_secs
60-
self.generic_data = generic_data
58+
self.reload_task = reload_task
59+
self.samples_per_plugin = samples_per_plugin or {}
60+
self.window_title = window_title
6161

6262

6363
class FakePlugin(base_plugin.TBPlugin):

tensorboard/backend/event_processing/data_ingester_test.py

Lines changed: 29 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -32,47 +32,51 @@
3232
class FakeFlags(object):
3333
def __init__(
3434
self,
35-
logdir,
35+
generic_data="auto",
36+
logdir="",
3637
logdir_spec="",
37-
purge_orphaned_data=True,
38-
reload_interval=60,
39-
samples_per_plugin=None,
4038
max_reload_threads=1,
41-
reload_task="auto",
42-
window_title="",
4339
path_prefix="",
40+
purge_orphaned_data=True,
41+
reload_interval=60,
4442
reload_multifile=False,
4543
reload_multifile_inactive_secs=4000,
46-
generic_data="auto",
44+
reload_task="auto",
45+
samples_per_plugin=None,
46+
window_title="",
4747
):
48+
self.generic_data = generic_data
4849
self.logdir = logdir
4950
self.logdir_spec = logdir_spec
50-
self.purge_orphaned_data = purge_orphaned_data
51-
self.reload_interval = reload_interval
52-
self.samples_per_plugin = samples_per_plugin or {}
5351
self.max_reload_threads = max_reload_threads
54-
self.reload_task = reload_task
55-
self.window_title = window_title
5652
self.path_prefix = path_prefix
53+
self.purge_orphaned_data = purge_orphaned_data
54+
self.reload_interval = reload_interval
5755
self.reload_multifile = reload_multifile
5856
self.reload_multifile_inactive_secs = reload_multifile_inactive_secs
59-
self.generic_data = generic_data
57+
self.reload_task = reload_task
58+
self.samples_per_plugin = samples_per_plugin or {}
59+
self.window_title = window_title
6060

6161

6262
class GetEventFileActiveFilterTest(tb_test.TestCase):
6363
def testDisabled(self):
64-
flags = FakeFlags("logdir", reload_multifile=False)
64+
flags = FakeFlags(logdir="logdir", reload_multifile=False)
6565
self.assertIsNone(data_ingester._get_event_file_active_filter(flags))
6666

6767
def testInactiveSecsZero(self):
6868
flags = FakeFlags(
69-
"logdir", reload_multifile=True, reload_multifile_inactive_secs=0
69+
logdir="logdir",
70+
reload_multifile=True,
71+
reload_multifile_inactive_secs=0,
7072
)
7173
self.assertIsNone(data_ingester._get_event_file_active_filter(flags))
7274

7375
def testInactiveSecsNegative(self):
7476
flags = FakeFlags(
75-
"logdir", reload_multifile=True, reload_multifile_inactive_secs=-1
77+
logdir="logdir",
78+
reload_multifile=True,
79+
reload_multifile_inactive_secs=-1,
7680
)
7781
filter_fn = data_ingester._get_event_file_active_filter(flags)
7882
self.assertTrue(filter_fn(0))
@@ -81,7 +85,9 @@ def testInactiveSecsNegative(self):
8185

8286
def testInactiveSecs(self):
8387
flags = FakeFlags(
84-
"logdir", reload_multifile=True, reload_multifile_inactive_secs=10
88+
logdir="logdir",
89+
reload_multifile=True,
90+
reload_multifile_inactive_secs=10,
8591
)
8692
filter_fn = data_ingester._get_event_file_active_filter(flags)
8793
with mock.patch.object(time, "time") as mock_time:
@@ -378,15 +384,19 @@ def testCheckFilesystemSupport_Called(self):
378384
with mock.patch.object(
379385
data_ingester, "_check_filesystem_support", autospec=True
380386
) as mock_check_filesystem_support:
381-
data_ingester.LocalDataIngester(flags=FakeFlags("logdir"))
387+
data_ingester.LocalDataIngester(
388+
flags=FakeFlags(logdir="logdir")
389+
)
382390
mock_check_filesystem_support.assert_called_once_with({"logdir"})
383391

384392
def testCheckFilesystemSupport_notCalled(self):
385393
with mock.patch.object(tf, "__version__", new="stub"):
386394
with mock.patch.object(
387395
data_ingester, "_check_filesystem_support", autospec=True
388396
) as mock_check_filesystem_support:
389-
data_ingester.LocalDataIngester(flags=FakeFlags("logdir"))
397+
data_ingester.LocalDataIngester(
398+
flags=FakeFlags(logdir="logdir")
399+
)
390400
mock_check_filesystem_support.assert_not_called()
391401

392402

tensorboard/plugins/core/core_plugin_test.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -47,30 +47,30 @@ class FakeFlags(object):
4747
def __init__(
4848
self,
4949
bind_all=False,
50+
db="",
51+
event_file="",
52+
generic_data="true",
53+
grpc_data_provider="",
5054
host=None,
5155
inspect=False,
52-
version_tb=False,
5356
logdir="",
5457
logdir_spec="",
55-
event_file="",
56-
db="",
5758
path_prefix="",
58-
generic_data="true",
59-
grpc_data_provider="",
6059
reuse_port=False,
60+
version_tb=False,
6161
):
6262
self.bind_all = bind_all
63+
self.db = db
64+
self.event_file = event_file
65+
self.generic_data = generic_data
66+
self.grpc_data_provider = grpc_data_provider
6367
self.host = host
6468
self.inspect = inspect
65-
self.version_tb = version_tb
6669
self.logdir = logdir
6770
self.logdir_spec = logdir_spec
68-
self.event_file = event_file
69-
self.db = db
7071
self.path_prefix = path_prefix
71-
self.generic_data = generic_data
72-
self.grpc_data_provider = grpc_data_provider
7372
self.reuse_port = reuse_port
73+
self.version_tb = version_tb
7474

7575

7676
class CorePluginFlagsTest(tf.test.TestCase):

0 commit comments

Comments
 (0)