| 313 | } |
| 314 | |
| 315 | func TestMap(t *testing.T) { |
| 316 | reader := FromSlice([]string{"hello", "world"}) |
| 317 | |
| 318 | lengths := Map(reader, func(s string) int { |
| 319 | return len(s) |
| 320 | }) |
| 321 | defer lengths.Close() |
| 322 | |
| 323 | result, err := lengths.Collect() |
| 324 | if err != nil { |
| 325 | t.Fatalf("unexpected error: %v", err) |
| 326 | } |
| 327 | |
| 328 | if len(result) != 2 || result[0] != 5 || result[1] != 5 { |
| 329 | t.Errorf("unexpected result: %v", result) |
| 330 | } |
| 331 | } |
| 332 | |
| 333 | func TestTake(t *testing.T) { |
| 334 | reader := FromSlice([]int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}) |