Skip to content

Commit edd5ad4

Browse files
kojix2kojix2
committed
Add linplot tests
1 parent 0bd2910 commit edd5ad4

3 files changed

Lines changed: 239 additions & 1 deletion

File tree

‎spec/reference_spec.cr‎

Lines changed: 222 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,14 @@ def densityplot_fixture_xprime(x : Array(Float64)) : Array(Float64)
3636
xprime
3737
end
3838

39+
def matrix_columns(rows : Array(Array(Float64))) : Array(Array(Float64))
40+
return [] of Array(Float64) if rows.empty?
41+
ncols = rows.first.size
42+
Array.new(ncols) do |col_idx|
43+
rows.map { |row| row[col_idx] }
44+
end
45+
end
46+
3947
describe "Julia reference output compatibility" do
4048
x = [-1.0, 1.0, 3.0, 3.0, -1.0]
4149
y = [2.0, 0.0, -5.0, 2.0, -5.0]
@@ -155,6 +163,166 @@ describe "Julia reference output compatibility" do
155163
test_ref("lineplot/sincostan2.txt", p)
156164
end
157165

166+
it "matches lineplot/sincos3" do
167+
p = UnicodePlot.lineplot(-0.5_f64, 3.0_f64, ->(v : Float64) { Math.sin(v) }, name: "sin(x)", ylabel: "f(x)", xlabel: "x")
168+
UnicodePlot.lineplot!(p, -0.5_f64, 3.0_f64, ->(v : Float64) { Math.cos(v) }, name: "cos(x)")
169+
test_ref("lineplot/sincos3.txt", p)
170+
end
171+
172+
it "matches lineplot/sin4" do
173+
tmp = [-0.5_f64, 0.6_f64, 1.4_f64, 2.5_f64]
174+
p = UnicodePlot.lineplot(tmp, ->(v : Float64) { Math.sin(v) }, name: "sin(x)", ylabel: "f(x)", xlabel: "x")
175+
test_ref("lineplot/sin4.txt", p)
176+
end
177+
178+
it "matches lineplot/sincos4" do
179+
tmp = [-0.5_f64, 0.6_f64, 1.4_f64, 2.5_f64]
180+
p = UnicodePlot.lineplot(tmp, ->(v : Float64) { Math.sin(v) }, name: "sin(x)", ylabel: "f(x)", xlabel: "x")
181+
UnicodePlot.lineplot!(p, tmp, tmp.map { |v| Math.cos(v) }, name: "cos(x)")
182+
test_ref("lineplot/sincos4.txt", p)
183+
end
184+
185+
it "matches lineplot/sincos_parameters" do
186+
p = UnicodePlot.lineplot(
187+
-0.5_f64,
188+
3.0_f64,
189+
->(v : Float64) { Math.sin(v) },
190+
name: "s",
191+
color: :red,
192+
title: "Funs",
193+
ylabel: "f",
194+
xlabel: "num",
195+
xlim: {-0.5_f64, 2.5_f64},
196+
ylim: {-0.9_f64, 1.2_f64},
197+
)
198+
UnicodePlot.lineplot!(p, -0.5_f64, 3.0_f64, ->(v : Float64) { Math.cos(v) }, name: "c", color: :yellow)
199+
test_ref("lineplot/sincos_parameters.txt", p)
200+
end
201+
202+
it "matches lineplot/slope1" do
203+
p = UnicodePlot.lineplot([2.0, 0.0, -5.0, 2.0, -5.0])
204+
UnicodePlot.lineplot!(p, -3, 1)
205+
test_ref("lineplot/slope1.txt", p)
206+
end
207+
208+
it "matches lineplot/slope2" do
209+
p = UnicodePlot.lineplot([2.0, 0.0, -5.0, 2.0, -5.0])
210+
UnicodePlot.lineplot!(p, -3, 1)
211+
UnicodePlot.lineplot!(p, -4, 0.5, color: :cyan, name: "foo")
212+
test_ref("lineplot/slope2.txt", p)
213+
end
214+
215+
it "matches lineplot/dates1" do
216+
xv = (730_119..730_149).map(&.to_f64)
217+
angles = Array.new(31) { |i| 3.0 * Math::PI * i / 30.0 }
218+
p = UnicodePlot.lineplot(xv, angles.map { |v| Math.sin(v) }, name: "sin", height: 5, xlabel: "date", xticks: false, xlim: {730_119.0, 730_149.0})
219+
p.label!(:bl, "1999-12-31")
220+
p.label!(:br, "2000-01-30")
221+
test_ref("lineplot/dates1.txt", p)
222+
end
223+
224+
pending "matches lineplot/dates2" do
225+
# Diff: exactly 1 glyph differs on the y=0 horizontal line (expected ⡤, got ⢤).
226+
# Cause: Julia uses Julian day integers directly as canvas coordinates.
227+
# cos(pi/2) = 6.12e-17 (floating-point rounding residual) causes the
228+
# DDA pixel_y value to straddle a floor() boundary by a tiny margin.
229+
# Depending on floating-point rounding direction, the braille off_col
230+
# bit becomes 0 or 1, producing the wrong dot. Julia's exact output
231+
# has been verified and matches the reference file.
232+
xv = (730_119..730_149).map(&.to_f64)
233+
angles = Array.new(31) { |i| 3.0 * Math::PI * i / 30.0 }
234+
p = UnicodePlot.lineplot(xv, angles.map { |v| Math.sin(v) }, name: "sin", height: 5, xlabel: "date", xticks: false, xlim: {730_119.0, 730_149.0}, ylim: {-1.0, 1.0})
235+
UnicodePlot.lineplot!(p, xv, angles.map { |v| Math.cos(v) }, name: "cos")
236+
p.label!(:bl, "1999-12-31")
237+
p.label!(:br, "2000-01-30")
238+
test_ref("lineplot/dates2.txt", p)
239+
end
240+
241+
it "matches lineplot/df1" do
242+
p = UnicodePlot.lineplot([0.0, 1.0, 2.0], [0.0, 1.0, -1.0], xticks: false)
243+
test_ref("lineplot/df1.txt", p)
244+
end
245+
246+
it "matches lineplot/df2" do
247+
p = UnicodePlot.lineplot([0.0, 1.0, 2.0], [0.0, 1.0, -1.0], xticks: false)
248+
p.label!(:bl, "2:3:0")
249+
p.label!(:br, "8:9:0")
250+
test_ref("lineplot/df2.txt", p)
251+
end
252+
253+
it "matches lineplot/color_vector" do
254+
p = UnicodePlot.lineplot([Float64::NAN], [Float64::NAN], xlim: {-1.0, 7.0}, ylim: {-1.0, 9.0})
255+
UnicodePlot.lineplot!(p, [-1.0, 2.0], [-1.0, 2.0], color: :red)
256+
UnicodePlot.lineplot!(p, [2.0, 3.0], [2.0, 9.0], color: :green)
257+
UnicodePlot.lineplot!(p, [3.0, 7.0], [9.0, 4.0], color: :blue)
258+
test_ref("lineplot/color_vector.txt", p)
259+
end
260+
261+
it "matches lineplot/matrix_auto" do
262+
xv = (0..10).map(&.to_f64)
263+
y1_rows = (0..10).map { |i| [-2.0 + i, 2.0 + i, 6.0 + i] }
264+
y2_rows = (0..10).map { |i| [6.0 - i, 18.0 - i] }
265+
y1_cols = matrix_columns(y1_rows)
266+
y2_cols = matrix_columns(y2_rows)
267+
p = UnicodePlot.lineplot([Float64::NAN], [Float64::NAN], xlim: {0.0, 10.0}, ylim: {-2.0, 16.0})
268+
UnicodePlot.lineplot!(p, xv, y1_cols[0], name: "y1")
269+
UnicodePlot.lineplot!(p, xv, y1_cols[1], name: "y2")
270+
UnicodePlot.lineplot!(p, xv, y1_cols[2], name: "y3")
271+
UnicodePlot.lineplot!(p, xv, y2_cols[0], name: "y1")
272+
UnicodePlot.lineplot!(p, xv, y2_cols[1], name: "y2")
273+
test_ref("lineplot/matrix_auto.txt", p)
274+
end
275+
276+
it "matches lineplot/matrix_parameters" do
277+
xv = (0..10).map(&.to_f64)
278+
y1_rows = (0..10).map { |i| [-2.0 + i, 2.0 + i, 6.0 + i] }
279+
y2_rows = (0..10).map { |i| [6.0 - i, 18.0 - i] }
280+
y1_cols = matrix_columns(y1_rows)
281+
y2_cols = matrix_columns(y2_rows)
282+
p = UnicodePlot.lineplot([Float64::NAN], [Float64::NAN], xlim: {0.0, 10.0}, ylim: {-2.0, 16.0})
283+
UnicodePlot.lineplot!(p, xv, y1_cols[0], name: "1", color: :red)
284+
UnicodePlot.lineplot!(p, xv, y1_cols[1], name: "2", color: :green)
285+
UnicodePlot.lineplot!(p, xv, y1_cols[2], name: "3", color: :blue)
286+
UnicodePlot.lineplot!(p, xv, y2_cols[0], name: "4", color: :yellow)
287+
UnicodePlot.lineplot!(p, xv, y2_cols[1], name: "5", color: :cyan)
288+
test_ref("lineplot/matrix_parameters.txt", p)
289+
end
290+
291+
pending "matches lineplot/intervalsets1" do
292+
# Diff: multiple glyphs differ across all rows.
293+
# Cause: Julia's IntervalSetsExt uses range(0..2; length=DEFAULT_WIDTH) which
294+
# samples DEFAULT_WIDTH (40) evenly-spaced points via a StepRange.
295+
# Crystal's lineplot(startx, endx, f) samples n = 3*width = 120 points
296+
# using startx + span * i / n, giving a higher point density.
297+
# The different segment lengths/slopes shift braille dot on/off states
298+
# throughout the curve. Julia's exact output has been verified to match
299+
# the reference file.
300+
p = UnicodePlot.lineplot(0.0_f64, 2.0_f64, ->(v : Float64) { v }, name: "identity(x)", xlabel: "x", ylabel: "f(x)", xlim: {0.0, 2.0}, ylim: {0.0, 2.0})
301+
UnicodePlot.lineplot!(p, 0.0_f64, 2.0_f64, ->(v : Float64) { Math.sqrt(v) }, name: "sqrt(x)")
302+
test_ref("lineplot/intervalsets1.txt", p)
303+
end
304+
305+
pending "matches lineplot/intervalsets2" do
306+
# Diff: multiple glyphs differ across all rows (same root cause as intervalsets1).
307+
# Cause: Julia samples range(0..1; length=DEFAULT_WIDTH) = 40 points.
308+
# Crystal samples n = 3*width = 120 points with startx + span * i / n,
309+
# so each DDA segment has a different slope and length, shifting braille
310+
# dot on/off states throughout the curve. Julia's exact output has been
311+
# verified to match the reference file.
312+
p = UnicodePlot.lineplot(0.0_f64, 1.0_f64, ->(v : Float64) { Math.sqrt(v) }, name: "sqrt(x)", xlabel: "x", ylabel: "f(x)", xlim: {0.0, 1.0}, ylim: {0.0, 1.0})
313+
UnicodePlot.lineplot!(p, 0.0_f64, 1.0_f64, ->(v : Float64) { v ** (1.0 / 3.0) }, name: "cbrt(x)")
314+
test_ref("lineplot/intervalsets2.txt", p)
315+
end
316+
317+
it "matches lineplot/units_pos_vel" do
318+
t = (0..100).map(&.to_f64)
319+
pos = t.map { |tv| 0.5 * tv * tv }
320+
vel = t
321+
p = UnicodePlot.lineplot(pos, vel, xlabel: "position (m)", ylabel: "speed (m s⁻¹)")
322+
UnicodePlot.lineplot!(p, [pos.min, pos.max], [vel.max, vel.max], color: :red)
323+
test_ref("lineplot/units_pos_vel.txt", p)
324+
end
325+
158326
it "matches lineplot/hvline" do
159327
p = UnicodePlot.lineplot([Float64::NAN], [Float64::NAN], xlim: {0.0, 8.0}, ylim: {0.0, 8.0})
160328
UnicodePlot.vline!(p, 2.0, [2.0, 6.0], color: :red)
@@ -165,6 +333,48 @@ describe "Julia reference output compatibility" do
165333
UnicodePlot.vline!(p, 1.0)
166334
test_ref("lineplot/hvline.txt", p)
167335
end
336+
337+
it "matches lineplot/ln_scale" do
338+
data = (10..1_000).step(10).map(&.to_f64).to_a
339+
p = UnicodePlot.lineplot(data, data, xscale: :ln, yscale: :ln)
340+
test_ref("lineplot/ln_scale.txt", p)
341+
end
342+
343+
it "matches lineplot/log2_scale" do
344+
data = (10..1_000).step(10).map(&.to_f64).to_a
345+
p = UnicodePlot.lineplot(data, data, xscale: :log2, yscale: :log2)
346+
test_ref("lineplot/log2_scale.txt", p)
347+
end
348+
349+
it "matches lineplot/log10_scale" do
350+
data = (10..1_000).step(10).map(&.to_f64).to_a
351+
p = UnicodePlot.lineplot(data, data, xscale: :log10, yscale: :log10)
352+
test_ref("lineplot/log10_scale.txt", p)
353+
end
354+
355+
it "matches lineplot/arrows" do
356+
p = UnicodePlot.lineplot([0.0, 1.0], [0.0, 1.0], head_tail: :head, name: "head", color: :red)
357+
UnicodePlot.lineplot!(p, [0.0, 1.0], [1.0, 0.0], head_tail: :tail, name: "tail", color: :green)
358+
UnicodePlot.lineplot!(p, [0.0, 1.0], [0.5, 0.5], head_tail: :both, name: "both", color: :blue)
359+
test_ref("lineplot/arrows.txt", p)
360+
end
361+
362+
it "matches lineplot/arrows_fractions" do
363+
n = 20
364+
xf = (0...n).map { |i| 1.0 + i.to_f / (n - 1) }
365+
p = UnicodePlot.lineplot(
366+
xf,
367+
Array.new(n, 0.0),
368+
ylim: {-1.0, 5.0},
369+
head_tail: :head,
370+
head_tail_frac: 0.05,
371+
name: "5%",
372+
)
373+
[{0.1, "10%"}, {0.15, "15%"}, {0.2, "20%"}, {0.25, "25%"}].each_with_index do |(frac, name), i|
374+
UnicodePlot.lineplot!(p, xf, Array.new(n, (i + 1).to_f), name: name, head_tail: :head, head_tail_frac: frac)
375+
end
376+
test_ref("lineplot/arrows_fractions.txt", p)
377+
end
168378
end
169379

170380
describe "stairs" do
@@ -205,6 +415,18 @@ describe "Julia reference output compatibility" do
205415
p = UnicodePlot.stairs([1.0, 2.0, 4.0, 7.0, 8.0], [1.0, 3.0, 4.0, 2.0, 7_000.0])
206416
test_ref("lineplot/stairs_edgecase.txt", p)
207417
end
418+
419+
it "matches lineplot/squeeze_annotations" do
420+
p = UnicodePlot.stairs(sx, sy, width: 20)
421+
p.label!(:tl, "Hello")
422+
p.label!(:t, "how are")
423+
p.label!(:tr, "you?")
424+
p.label!(:bl, "Hello")
425+
p.label!(:b, "how are")
426+
p.label!(:br, "you?")
427+
UnicodePlot.lineplot!(p, 1, 0.5)
428+
test_ref("lineplot/squeeze_annotations.txt", p)
429+
end
208430
end
209431

210432
describe "scatterplot" do

‎src/unicode_plot/interface/lineplot.cr‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,18 @@ module UnicodePlot
142142
lineplot!(plot, x, to_plot_f64(y), **kwargs)
143143
end
144144

145+
def lineplot!(plot : Plot, intercept : Number, slope : Number, **kwargs) : Plot
146+
can = plot.canvas
147+
xmin = can.origin_x
148+
xmax = can.origin_x + can.width
149+
intercept_f = intercept.to_f64
150+
slope_f = slope.to_f64
151+
lineplot!(plot,
152+
[xmin, xmax],
153+
[intercept_f + xmin * slope_f, intercept_f + xmax * slope_f],
154+
**kwargs)
155+
end
156+
145157
def lineplot!(plot : Plot, f : Float64 -> Float64, **kwargs) : Plot
146158
can = plot.canvas
147159
n = 3 * can.ncols

‎src/unicode_plot/show.cr‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,11 @@ module UnicodePlot
177177
end
178178

179179
# Print top labels (:t, :tl, :tr)
180-
print_labels(io, p, :t, nc - 2, border_left_pad + blank_in.to_s, blank_in.to_s + " " * max_len_r, blank_in, use_color) if has_labels
180+
top_label_rows = 0
181+
if has_labels
182+
top_label_rows = print_labels(io, p, :t, nc - 2, border_left_pad + blank_in.to_s, blank_in.to_s + " " * max_len_r, blank_in, use_color)
183+
io << '\n' if top_label_rows > 0
184+
end
181185

182186
# Print top border
183187
if g.visible?

0 commit comments

Comments
 (0)