Begin to work on conditions

Author: jecisc

src/FAST-Python-Tools-Tests/FASTPythonCFGTest.class.st

@@ -31,6 +31,29 @@ FASTPythonCFGTest >> setUp [
 	builder := FASTPythonCFGBuilder new
 ]
 
+{ #category : 'tests' }
+FASTPythonCFGTest >> testFunctionWithIf [
+
+	| nextBlock |
+	self skip.
+	self flag: #todo. "Finish"
+	self buildCFGFor: 'def f(i):
+    if i > 3:
+        print(i)'.
+
+	self assert: startBlock isConditional.
+	self assert: startBlock isFinal. "If the condition is false then the block is indeed final"
+	self assert: startBlock statements size equals: 1.
+	self assert: (startBlock statements first class isOfType: FASTPyComparisonOperator).
+	self assert: startBlock nextBlocks size equals: 2.
+	self assert: (startBlock nextBlock select: #isNullBlock) size equals: 1.
+
+	nextBlock := startBlock nextBlockForValue: true.
+	self assert: nextBlock isConditional not.
+	self assert: nextBlock isFinal.
+	self assertEmpty: nextBlock statements
+]
+
 { #category : 'tests' }
 FASTPythonCFGTest >> testFunctionWithOneStatement [
 

src/FAST-Python-Tools/FASTPythonCFGVisitor.class.st

@@ -35,6 +35,14 @@ FASTPythonCFGVisitor >> visitFASTPyReturnStatement: aReturn [
 	cfgBuilder currentBlock nextBlock: FASTNullBlock new
 ]
 
+{ #category : 'visiting' }
+FASTPythonCFGVisitor >> visitFASTTConditionalStatement: aStatement [
+
+	cfgBuilder newConditionalBlock: aStatement.
+
+	super visitFASTTConditionalStatement: aStatement
+]
+
 { #category : 'visiting' }
 FASTPythonCFGVisitor >> visitFASTTStatement: aStatement [