Assertion Roulette::Detecting AssertCount in each method

detecting_assertion_roulette.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+# In[1]:
+
+
+import ast
+from pprint import pprint
+import inspect
+import astor
+
+
+# In[67]:
+
+
+def main():
+    print(detectAssertionRoulette("test_code_for_parsing.py"))
+
+
+# In[68]:
+
+
+def detectAssertionRoulette(script_name):
+# Defining Array to hold the function name, count of Assert statements & presence of smell
+    fns = []
+
+    #parsing the script using AST
+    with open(script_name, "r") as source:
+        tree = ast.parse(source.read())
+        for block in tree.body:
+            for node in ast.walk(block):
+                fn = {}
+                #checking whether current node is a function
+                if isinstance(node, ast.FunctionDef):
+                    fn['name'] = node.name
+#                     fn['body'] = astor.to_source(node)
+                    fn['countAssert'] = 0
+                    fn['hasSmell'] = False
+                    for body_item in node.body:
+                        #checking if the function body has any assert statement
+                        #and keeping the count of that
+                        if isinstance(body_item, ast.Assert):
+                            fn['countAssert'] = fn['countAssert'] + 1
+
+                    if fn['countAssert'] >1:
+                        fn['hasSmell'] = True
+
+
+                    fns.append(fn)
+        return fns
+
+
+# In[71]:
+
+
+if __name__ == "__main__":
+    main()
+
+
+# In[ ]:
+
+
+#https://stackoverflow.com/questions/54092879/how-to-find-out-if-the-source-code-of-a-function-contains-a-loop
+