dart-sdk/tests/language_2/async_star/take_reyield_test.dart
Robert Nystrom fc1b1ecc71 Move files under language_2 into subdirectories.
Change-Id: Idbcc965a27e9ffeedf5e0a1068b019de4193070f
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/127745
Commit-Queue: Bob Nystrom <rnystrom@google.com>
Reviewed-by: Alexander Thomas <athom@google.com>
2019-12-11 19:18:00 +00:00

29 lines
757 B
Dart

// Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
import "dart:async";
import "package:expect/expect.dart";
import "package:async_helper/async_helper.dart";
expectList(actualStream, expectedList) {
return actualStream.toList().then((v) {
Expect.listEquals(expectedList, v);
});
}
Stream makeStream(int n) async* {
for (int i = 0; i < n; i++) yield i;
}
main() {
fivePartialSums(Stream s) async* {
var r = 0;
await for (var v in s.take(5)) yield r += v;
}
asyncStart();
expectList(fivePartialSums(makeStream(10)), [0, 1, 3, 6, 10])
.then(asyncSuccess);
}