From 0bd6567a77d97e24874d658e0067790a9ff690e1 Mon Sep 17 00:00:00 2001
From: Jonathan Cave <jonathan.cave@canonical.com>
Date: Tue, 19 Jun 2018 16:15:37 +0100
Subject: Remove argv passing, neater argument check

---
 bin/fde_tests.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/bin/fde_tests.py b/bin/fde_tests.py
index 92a6cc2..3008735 100755
--- a/bin/fde_tests.py
+++ b/bin/fde_tests.py
@@ -21,10 +21,8 @@ import subprocess as sp
 import sys
 
 
-def main(argv):
-    on_desktop = False
-    if len(sys.argv) > 1 and sys.argv[1] == 'desktop':
-        on_desktop = True
+def main():
+    on_desktop = len(sys.argv) > 1 and sys.argv[1] == 'desktop'
 
     # the mountpoint corresponding to the on disk encrypted partition
     base_mount = '/' if on_desktop else '/writable'
@@ -100,4 +98,4 @@ def main(argv):
 
 
 if __name__ == "__main__":
-    main(sys.argv)
+    main()
-- 
cgit v1.2.3