diff --git a/ds/dataset.go b/ds/dataset.go index 64b9976..fed1802 100644 --- a/ds/dataset.go +++ b/ds/dataset.go @@ -248,9 +248,9 @@ func (d *Dataset) GetBody(thread *starlark.Thread, _ *starlark.Builtin, args sta if err != nil { return starlark.None, err } - provider.SetBodyFile(qfs.NewMemfileBytes("data.json", data)) + provider.SetBodyFile(qfs.NewMemfileBytes("body.json", data)) - rr, err := dsio.NewEntryReader(provider.Structure, qfs.NewMemfileBytes("data.json", data)) + rr, err := dsio.NewEntryReader(provider.Structure, qfs.NewMemfileBytes("body.json", data)) if err != nil { return starlark.None, fmt.Errorf("error allocating data reader: %s", err) } @@ -310,7 +310,7 @@ func (d *Dataset) SetBody(thread *starlark.Thread, _ *starlark.Builtin, args sta return starlark.None, fmt.Errorf("expected data for '%s' format to be a string", df) } - d.write.SetBodyFile(qfs.NewMemfileBytes(fmt.Sprintf("data.%s", df), []byte(string(str)))) + d.write.SetBodyFile(qfs.NewMemfileBytes(fmt.Sprintf("body.%s", df), []byte(string(str)))) d.modBody = true d.bodyCache = nil return starlark.None, nil diff --git a/ds/dataset_test.go b/ds/dataset_test.go index de1a89e..5be0ecf 100644 --- a/ds/dataset_test.go +++ b/ds/dataset_test.go @@ -228,7 +228,7 @@ bat,3,meh }, }, } - ds.SetBodyFile(qfs.NewMemfileBytes("data.csv", []byte(text))) + ds.SetBodyFile(qfs.NewMemfileBytes("body.csv", []byte(text))) d := NewDataset(ds, nil) d.SetMutable(&dataset.Dataset{ diff --git a/ds/testdata/test.star b/ds/testdata/test.star index a1e3d7a..3ff66b0 100644 --- a/ds/testdata/test.star +++ b/ds/testdata/test.star @@ -38,11 +38,11 @@ assert.eq(ds.set_body("[[1,2,3]]", parse_as="json"), None) assert.eq(ds.get_body(), bd) # csv_ds is a global variable provided by dataset_test.go -# "cycling" csv data through starlark shouldn't have significant effects on the +# round-tripping csv data through starlark shouldn't have significant effects on the # encoded data. whitespace is *not* significant. # csv data is one of the harder formats, where there header row must be preserved csv_ds.set_body(csv_ds.get_body()) expect_data = [["foo",1,"true"], ["bar",2,"false"], ["bat",3,"meh"]] assert.eq(expect_data, csv_ds.get_body()) -assert.eq(csv_ds.get_structure()['format'], 'csv') \ No newline at end of file +assert.eq(csv_ds.get_structure()['format'], 'csv')