Tricount- getting WA in python

[ninety]

t=int(raw_input())
for i in range(t):
	n=long(raw_input())
	if n&(n-1)==0:
		print long((n*(n+2)*(2*n+1))/8)
	else:
		print long((n*(n+2)*(2*n+1)-1)/8)

I wrote this code and I got a WA. But same code in c++ got accepted. How should I increase the range of solution?

i later edited it to

t=int(raw_input())
for i in range(t):
	n=long(raw_input())
	if n&(n-1)==0:
		k= long((n*(n+2)*(2*n+1))/8)
	else:
		k= long((n*(n+2)*(2*n+1)-1)/8)
	if k<0:
		k=k&0xffffffffffffffff
	print k

it passed the 5th test but gave wa on 8.