Skip to content

Commit 79f4e7f

Browse files
committed
changed output_collector key name
1 parent 5f534a9 commit 79f4e7f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dsi/plugins/tests/test_env.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,24 +34,24 @@ def test_hostname_plugin_row_shape():
3434
# SYSTEM KERNEL FUNCTIONS ONLY WORK ON LINUX
3535
def test_systemkernel_plugin_type():
3636
plug = SystemKernel()
37-
assert type(plug.output_collector["SystemKernel"]) == collections.OrderedDict
37+
assert type(plug.output_collector) == collections.OrderedDict
3838

3939
def test_systemkernel_plugin_adds_rows():
4040
plug = SystemKernel()
4141
plug.add_rows()
4242
plug.add_rows()
4343

44-
for key, val in plug.output_collector["SystemKernel"].items():
44+
for key, val in plug.output_collector.items():
4545
assert len(val) == 2
4646

4747
# 1 SystemKernel column + 4 inherited Env cols
48-
assert len(plug.output_collector["SystemKernel"].keys()) == 5
48+
assert len(plug.output_collector.keys()) == 5
4949

5050
def test_systemkernel_plugin_blob_is_big():
5151
plug = SystemKernel()
5252
plug.add_rows()
5353

54-
blob = plug.output_collector["SystemKernel"]["kernel_info"][0]
54+
blob = plug.output_collector["kernel_info"][0]
5555
info_dict = loads(blob)
5656

5757
# dict should have more than 1000 (~7000) keys

0 commit comments

Comments
 (0)