can exit while reading from keyboard
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package input_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"mayhem-party/src/device/input"
|
||||
"os"
|
||||
"path"
|
||||
@@ -8,7 +9,7 @@ import (
|
||||
)
|
||||
|
||||
func TestNew(t *testing.T) {
|
||||
r := input.New()
|
||||
r := input.New(context.Background())
|
||||
for i := 0; i < 15; i++ {
|
||||
batch := r.Read()
|
||||
for _, got := range batch {
|
||||
@@ -23,7 +24,7 @@ func TestNewBuffered(t *testing.T) {
|
||||
os.Unsetenv("INPUT_BUFFERED")
|
||||
})
|
||||
|
||||
r := input.New()
|
||||
r := input.New(context.Background())
|
||||
for i := 0; i < 15; i++ {
|
||||
batch := r.Read()
|
||||
for j := range batch {
|
||||
@@ -44,7 +45,7 @@ func TestNewRandomWeightFile(t *testing.T) {
|
||||
os.Unsetenv("INPUT_RANDOM_WEIGHT_FILE")
|
||||
})
|
||||
|
||||
r := input.New()
|
||||
r := input.New(context.Background())
|
||||
for i := 0; i < 15; i++ {
|
||||
batch := r.Read()
|
||||
for _, got := range batch {
|
||||
|
||||
Reference in New Issue
Block a user