480_setcomp.py 287 B

1234567891011
  1. a = {i for i in range(10)}
  2. assert a == set(range(10))
  3. a = {i for i in range(10) if i % 2 == 0}
  4. assert a == {0, 2, 4, 6, 8}
  5. a = {i**3 for i in range(10) if i % 2 == 0}
  6. assert a == {0, 8, 64, 216, 512}
  7. a = {(i,i+1) for i in range(5)}
  8. assert a == {(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)}