refactor to satisfy flak8

This commit is contained in:
scito
2022-09-03 16:12:28 +02:00
committed by Roland Kurmann
parent 7f8250c0a6
commit 175e7c38e5
5 changed files with 37 additions and 8 deletions

View File

@@ -25,6 +25,7 @@ import os
import extract_otp_secret_keys
class TestExtract(unittest.TestCase):
def test_extract_csv(self):
@@ -53,9 +54,11 @@ class TestExtract(unittest.TestCase):
remove_file('test_example_output.csv')
remove_file('test_example_output.json')
def remove_file(filename):
if os.path.exists(filename): os.remove(filename)
def read_csv(filename):
"""Returns a list of lines."""
with open(filename, "r") as infile:
@@ -65,6 +68,7 @@ def read_csv(filename):
lines.append(line)
return lines
def read_json(filename):
"""Returns a list or a dictionary."""
with open(filename, "r") as infile: