Changeset 23793 for issm/trunk-jpl/test/NightlyRun/GetIds.py

Timestamp:

03/13/19 03:17:46 (6 years ago)

Author:

bdef

Message:

pep8 compliance of NTs

File:

• issm/trunk-jpl/test/NightlyRun/GetIds.py (modified) (1 diff)

issm/trunk-jpl/test/NightlyRun/GetIds.py

@@ -4 +4 @@
 import numpy as np
 
+
 def GetIds(ids_names):
-        """
-        GetIds - output ids from a given array of IDs and test names
+    """
+     GetIds - output ids from a given array of IDs and test names
 
-                 the test names can be any string or sub-string present
-                 in the test's name (first line of corresponding file)
+              the test names can be any string or sub-string present
+              in the test's name (first line of corresponding file)
 
-                 test names are case sensitive
- 
-            Usage:
-                ids=GetIds(101)
-                ids=GetIds('Dakota')
-                ids=GetIds([101,102...])
-                ids=GetIds([\'Dakota\',\'Slr\'...])
-                ids=GetIds([[101,102...],[\'Dakota\',\'Slr\'...]])
-        """
+              test names are case sensitive
 
-        ids = []
+         Usage:
+             ids = GetIds(101)
+             ids = GetIds('Dakota')
+             ids = GetIds([101, 102...])
+             ids = GetIds([\'Dakota\',\'Slr\'...])
+             ids = GetIds([[101, 102...],[\'Dakota\',\'Slr\'...]])
+    """
 
-        # 1 input, either an id or a test name
-        if type(ids_names) == str:
-                ids = IdFromString(ids_names)
-                if len(ids) == 0:
-                        # fail silently
-                        return []
-                        #raise RuntimeError('runme.py: GetIds.py: No tests with names matching "'+ids_names+'" were found. Note that name checking is case sensitive. Test names are in the first line of a given test eg: "Square" would include test101.py: "SquareShelfConstrainedStressSSA2d"')
+    ids = []
 
-        if type(ids_names) == int:
-                ids = [ids_names]
-                if len(ids) == 0:
-                        # fail silently
-                        return []
-                        #raise RuntimeError('runme.py: GetIds.py: No tests with ids matching "'+ids_names+'" were found. Check that there is a test file named "test'+str(ids_names)+'.py"')
+    # 1 input, either an id or a test name
+    if type(ids_names) == str:
+        ids = IdFromString(ids_names)
+        if len(ids) == 0:
+            # fail silently
+            return []
+        #raise RuntimeError('runme.py: GetIds.py: No tests with names matching "'+ids_names+'" were found. Note that name checking is case sensitive. Test names are in the first line of a given test eg: "Square" would include test101.py: "SquareShelfConstrainedStressSSA2d"')
 
-        # many inputs of either ids or test names
-        if type(ids_names) == list and len(ids_names) > 0:
-                # is everything a string or int?
-                if np.array([type(i) == int for i in ids_names]).all():
-                        ids = ids_names
-                elif np.array([type(i) == str for i in ids_names]).all():
-                        ids = np.concatenate([IdFromString(i) for i in ids_names])
-                        if len(ids) == 0:
-                                raise RuntimeError('runme.py: GetIds.py: No tests with names matching "'+ids_names+'" were found. Note that name checking is case sensitive.')
+    if type(ids_names) == int:
+        ids = [ids_names]
+        if len(ids) == 0:
+            # fail silently
+            return []
+        #raise RuntimeError('runme.py: GetIds.py: No tests with ids matching "'+ids_names+'" were found. Check that there is a test file named "test'+str(ids_names)+'.py"')
 
-        # many inputs of both ids and test names
-        # ids_names[0] -> ids_names by id
-        # ids_names[1] -> ids_names by test name
-        if type(ids_names) == list and len(ids_names) == 2:
-                if type(ids_names[0]) == list and len(ids_names[0]) > 0 and type(ids_names[0][0]) == int:
-                        ids = np.concatenate([ids,ids_names[0]])
-                if type(ids_names[1]) == list and len(ids_names[1]) > 0 and type(ids_names[1][0]) == str:
-                        ids = np.concatenate([ids,np.concatenate([IdFromString(i) for i in ids_names[1]])])
-                        if len(ids) == 0:
-                                raise RuntimeError('runme.py: GetIds.py: No tests with names matching "'+ids_names+'" were found. Note that name checking is case sensitive.')
+    # many inputs of either ids or test names
+    if type(ids_names) == list and len(ids_names) > 0:
+        # is everything a string or int?
+        if np.array([type(i) == int for i in ids_names]).all():
+            ids = ids_names
+        elif np.array([type(i) == str for i in ids_names]).all():
+            ids = np.concatenate([IdFromString(i) for i in ids_names])
+            if len(ids) == 0:
+                raise RuntimeError('runme.py: GetIds.py: No tests with names matching "' + ids_names + '" were found. Note that name checking is case sensitive.')
 
-        # no recognizable ids or id formats
-        if np.size(ids) == 0 and not np.all(np.equal(ids_names,None)):
-                raise RuntimeError('runme.py: GetIds.py: include and exclude options (-i/--id; -in/--include_name; -e/--exclude; -en/--exclude_name) options must follow GetIds usage format:\n'+GetIds.__doc__)
+    # many inputs of both ids and test names
+    # ids_names[0] -> ids_names by id
+    # ids_names[1] -> ids_names by test name
+    if type(ids_names) == list and len(ids_names) == 2:
+        if type(ids_names[0]) == list and len(ids_names[0]) > 0 and type(ids_names[0][0]) == int:
+            ids = np.concatenate([ids, ids_names[0]])
+        if type(ids_names[1]) == list and len(ids_names[1]) > 0 and type(ids_names[1][0]) == str:
+            ids = np.concatenate([ids, np.concatenate([IdFromString(i) for i in ids_names[1]])])
+            if len(ids) == 0:
+                raise RuntimeError('runme.py: GetIds.py: No tests with names matching "' + ids_names + '" were found. Note that name checking is case sensitive.')
 
-        return np.array(ids).astype(int)
+    # no recognizable ids or id formats
+    if np.size(ids) == 0 and not np.all(np.equal(ids_names, None)):
+        raise RuntimeError('runme.py: GetIds.py: include and exclude options (-i/--id; -in/--include_name; -e/--exclude; -en/--exclude_name) options must follow GetIds usage format:\n' + GetIds.__doc__)
 
+    return np.array(ids).astype(int)