Shorthand return-expression

dlepikhova · dlepikhova · commit 94caeb1793fe · 2022-07-08T15:51:09.000+05:00
diff --git a/tests/helpers/ptrack_helpers.py b/tests/helpers/ptrack_helpers.py
@@ -100,11 +100,7 @@ def is_enterprise():
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE
     )
-    if b'enterprise' in p.communicate()[0]:
-        return True
-    else:
-        return False
-
+    return b'postgrespro.ru' in p.communicate()[0]
 
 class ProbackupException(Exception):
     def __init__(self, message, cmd):
