from arraybag import ArrayBag

from ball import Ball

from linkedbag import LinkedBag

from node import Node

# Part $1$:

def distributeBag$($bag$)$:

redBag $=$ ArrayBag$()$

blueBag $=$ ArrayBag$()$

# Iterate through the original bag:

for ball in bag:

# Inflate red balls and add them to the red bag:

if ball.getColor$()==$ "red":

ball.setRadius$($ball$.$getRadius$()*2)$

redBag.add$($ball$)$

else:

# Add blue balls to the blue bag:

blueBag.add$($ball$)$

# Empty the original bag:

bag.clear$()$

return redBag, blueBag

# Part $2$:

def printBag$($bag$)$:

if bag.isEmpty$()$:

print$("$empty$")$

else:

for item in bag:

print$($item$)$

print$()$ # Print a blank line after the contents

# Part $3$:

def removeDuplicates$($bag$)$:

while True:

count $=0$

for item in bag:

count $+=1$

if count $>1$:

bag.remove$($item$)$

break

else:

return

# Test Cases:

# Test $1$:

print$("$Test $1$:$")$

bag$1=$ ArrayBag$([$Ball$("$red$",10),$

Ball$("$red$",11),$

Ball$("$red$",12),$

Ball$("$blue$",15),$

Ball$("$blue$",15),$

Ball$("$blue$",15)])$

print$("$Original mixed bag:"$)$

printBag$($bag$1)$

redBag, blueBag $=$ distributeBag$($bag$1)$

print$("$Red bag:"$)$

printBag$($redBag$)$

print$("$Blue bag:"$)$

printBag$($blueBag$)$

print$("$Final mixed bag:"$)$

printBag$($bag$1)$

# Test $2$:

print$("$Test $2$:$")$

bag$2=$ ArrayBag$([$Ball$("$red$",201),$

Ball$("$red$",201),$

Ball$("$red$",202),$

Ball$("$red$",202)])$

print$("$Original mixed bag:"$)$

printBag$($bag$2)$

redBag, blueBag $=$ distributeBag$($bag$2)$

print$("$Red bag:"$)$

printBag$($redBag$)$

print$("$Blue bag:"$)$

printBag$($blueBag$)$

print$("$Final mixed bag:"$)$

printBag$($bag$2)$

# Test $3$:

print$("$Test $3$:$")$

bag$3=$ ArrayBag$()$

print$("$Original mixed bag:"$)$

printBag$($bag$3)$

redBag, blueBag $=$ distributeBag$($bag$3)$

print$("$Red bag:"$)$

printBag$($redBag$)$

print$("$Blue bag:"$)$

printBag$($blueBag$)$

print$("$Final mixed bag:"$)$

printBag$($bag$3)$

# Test $4$:

print$("$Test $4$:$")$

bag$4=$ LinkedBag$(["$apple$",$

"apple",

"banana",

"kiwi",

"cantaloupe",

"pear",

"banana",

"orange",

"orange",

"cantaloupe",

"apple",

"lemon",

"lime",

"lime"$])$

print$("$Original bag with duplicates:"$)$

printBag$($bag$4)$

removeDuplicates$($bag$4)$

print$("$Bag after removing duplicates:"$)$

printBag$($bag$4)$

I need to know what the problem is and how fix this