Test generate and replace without @with

codedthinking · Sep 19, 2024 · bb9862e · bb9862e
1 parent 61e2e96
commit bb9862e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/docs/examples/benchmark.jl b/docs/examples/benchmark.jl
@@ -6,11 +6,17 @@ using Pkg; Pkg.precompile()
 df = DataFrame(i = 1:10_000_000)
 df.g = rand(0:99, nrow(df))
 
+
 println("Generate")
-@btime @with df @generate ln_i = log(i)
+setdf(df)
+@time @generate ln_i = log(i)
+setdf(df)
+@time @generate ln_i = log(i)
+
+setdf(df)
 
 println("Replace")
-@btime @with df @replace g = 2*i
+@btime @replace g = 2*i
 
 println("Egen")
 @btime @with df  @egen mean_i = mean(i), by(g)

diff --git a/docs/src/index.md b/docs/src/index.md
@@ -68,7 +68,7 @@ end
 | Command      | Stata  | Julia 2nd run | Speedup |
 | ------------ | -----  | ------------- | ------- |
 | `@generate`  | 230ms  | 46ms | 5x |
-| `@replace`   | 232ms  | 43ms | 5x |
+| `@replace`   | 232ms  | 32ms | 7x |
 | `@egen`      | 5.00s  | 0.37s         | 13x     |
 | `@collapse`  | 0.94s  | 0.28s         | 3x      |
 | `@tabulate`  | 2.19s  | 0.09s         | 24x     |