Assignment in assignment

Author: jecisc

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

@@ -17,6 +17,51 @@ FASTPythonSSATest >> parseAndResolve: aString [
 	FASTPythonSSAVisitor resolve: model module
 ]
 
+{ #category : 'tests - assignments' }
+FASTPythonSSATest >> testAssignementInAssignment [
+
+	| assignedVar1 assignedVar2 assignedVar3 usedVar1 usedVar2 usedVar3 usedVar4 |
+	self parseAndResolve: 'x = y = 3
+
+print(x)
+print(y)
+
+x = 5
+
+print(x)
+print(y)'.
+
+	assignedVar1 := model module statements first left.
+	self assert: assignedVar1 ssaVersion version equals: 1.
+	self assert: assignedVar1 ssaName equals: 'x_1'.
+	self assert: assignedVar1 ssaVersion localDeclaration equals: assignedVar1 localDeclaration.
+	self assert: assignedVar1 ssaVersion localUses size equals: 2.
+
+	assignedVar2 := model module statements first right left.
+	self assert: assignedVar2 ssaVersion version equals: 1.
+	self assert: assignedVar2 ssaName equals: 'y_1'.
+	self assert: assignedVar2 ssaVersion localDeclaration equals: assignedVar2 localDeclaration.
+	self assert: assignedVar2 ssaVersion localUses size equals: 3.
+
+	usedVar1 := model module statements second arguments first.
+	self assert: usedVar1 ssaVersion identicalTo: assignedVar1 ssaVersion.
+
+	usedVar2 := model module statements third arguments first.
+	self assert: usedVar2 ssaVersion identicalTo: assignedVar2 ssaVersion.
+
+	assignedVar3 := model module statements fourth left.
+	self assert: assignedVar3 ssaVersion version equals: 2.
+	self assert: assignedVar3 ssaName equals: 'x_2'.
+	self assert: assignedVar3 ssaVersion localDeclaration equals: assignedVar3 localDeclaration.
+	self assert: assignedVar3 ssaVersion localUses size equals: 2.
+
+	usedVar3 := model module statements fifth arguments first.
+	self assert: usedVar3 ssaVersion identicalTo: assignedVar3 ssaVersion.
+
+	usedVar4 := model module statements sixth arguments first.
+	self assert: usedVar4 ssaVersion identicalTo: assignedVar2 ssaVersion
+]
+
 { #category : 'tests - assignments' }
 FASTPythonSSATest >> testAssignementInIfInBothBranchesAndUsage [