formatting

davidbailey · Dec 9, 2019 · 09e3e39 · 09e3e39
1 parent 9a96de1
commit 09e3e39
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 11 deletions.
diff --git a/rail/cpi.py b/rail/cpi.py
@@ -12,7 +12,7 @@ class CPI:
     def __init__(self) -> None:
         url = "https://download.bls.gov/pub/time.series/cu/cu.data.0.Current"
         self.cpi = pd.read_csv(url, sep="\t", skipinitialspace=True)
-        self.cpi.columns = [c.replace(' ', '') for c in self.cpi.columns]
+        self.cpi.columns = [c.replace(" ", "") for c in self.cpi.columns]
 
     def inflation(self, from_year: int, to_year: int) -> float:
         """

diff --git a/tests/test_vulnerability.py b/tests/test_vulnerability.py
@@ -31,17 +31,16 @@ def test_vulnerabilities(self):
         """
         Test vulnerabilities
         """
-        name = self.threat_event["name"] + " -> " + self.system.path() + " | " + str(self.controls)
-        self.assertEqual(
-            self.vulnerabilities[name]["threat_event"],
-            self.threat_event,
-        )
-        self.assertEqual(
-            self.vulnerabilities[name]["system"], self.system
-        )
-        self.assertEqual(
-            self.vulnerabilities[name]["controls"], self.controls
+        name = (
+            self.threat_event["name"]
+            + " -> "
+            + self.system.path()
+            + " | "
+            + str(self.controls)
         )
+        self.assertEqual(self.vulnerabilities[name]["threat_event"], self.threat_event)
+        self.assertEqual(self.vulnerabilities[name]["system"], self.system)
+        self.assertEqual(self.vulnerabilities[name]["controls"], self.controls)
 
 
 if __name__ == "__main__":