Skip to content

Commit 7397e98

Browse files
committed
Refactor test
1 parent 858671e commit 7397e98

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

test/mudbrick/parser/roundtrip_test.exs

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,7 @@ defmodule Mudbrick.ParseRoundtripTest do
1616
images = Keyword.get(document_options, :images, %{})
1717

1818
input =
19-
Enum.zip([
20-
pages_options,
21-
if(Enum.empty?(images_options), do: [], else: Stream.cycle(images_options)),
22-
if(Enum.empty?(images), do: [], else: Stream.cycle(Map.keys(images)))
23-
])
19+
Enum.zip([pages_options, cycle(images_options), cycle(Map.keys(images))])
2420
|> Enum.reduce(doc, fn {page_options, image_options, image_identifier}, context ->
2521
context
2622
|> page(page_options)
@@ -188,4 +184,7 @@ defmodule Mudbrick.ParseRoundtripTest do
188184

189185
assert parsed == input
190186
end
187+
188+
defp cycle([]), do: []
189+
defp cycle(enumerable), do: Stream.cycle(enumerable)
191190
end

0 commit comments

Comments
 (0)