github.com/sieve-data/cog@v0.0.72/pkg/cli/init_test.go (about)

     1  package cli
     2  
     3  import (
     4  	"os"
     5  	"path"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/require"
     9  )
    10  
    11  func TestInit(t *testing.T) {
    12  	dir := t.TempDir()
    13  
    14  	require.NoError(t, os.Chdir(dir))
    15  
    16  	err := initCommand([]string{})
    17  	require.NoError(t, err)
    18  
    19  	require.FileExists(t, path.Join(dir, "cog.yaml"))
    20  	require.FileExists(t, path.Join(dir, "predict.py"))
    21  }