mirror of https://github.com/databricks/cli.git
25 lines
571 B
Go
25 lines
571 B
Go
|
package python
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFreeze(t *testing.T) {
|
||
|
|
env, err := Freeze(context.Background())
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Greater(t, len(env), 1)
|
||
|
|
assert.True(t, env.Has("urllib3"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPyInlineX(t *testing.T) {
|
||
|
|
defer chdirAndBack("internal/test/simple-python-wheel")()
|
||
|
|
dist, err := ReadDistribution(context.Background())
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "dummy", dist.Name)
|
||
|
|
assert.Equal(t, "dummy", dist.Packages[0])
|
||
|
|
assert.True(t, dist.InstallEnvironment().Has("requests"))
|
||
|
|
}
|