| 294 | } |
| 295 | |
| 296 | func TestFilter(t *testing.T) { |
| 297 | reader := FromSlice([]int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}) |
| 298 | |
| 299 | filtered := Filter(reader, func(v int) bool { |
| 300 | return v > 5 |
| 301 | }) |
| 302 | defer filtered.Close() |
| 303 | |
| 304 | result, err := filtered.Collect() |
| 305 | if err != nil { |
| 306 | t.Fatalf("unexpected error: %v", err) |
| 307 | } |
| 308 | |
| 309 | expected := []int{6, 7, 8, 9, 10} |
| 310 | if len(result) != len(expected) { |
| 311 | t.Fatalf("expected %d items, got %d", len(expected), len(result)) |
| 312 | } |
| 313 | } |
| 314 | |
| 315 | func TestMap(t *testing.T) { |
| 316 | reader := FromSlice([]string{"hello", "world"}) |