Add some extra tests for take_every and drop_every (#4702)

Also, now testing the correct function in the negative test.

2 files changed, 11 insertions, 1 deletions

diff --git a/lib/elixir/test/elixir/enum_test.exs b/lib/elixir/test/elixir/enum_test.exs
index ad6bd6969..7902af2fa 100644
--- a/lib/elixir/test/elixir/enum_test.exs
+++ b/lib/elixir/test/elixir/enum_test.exs
@@ -126,6 +126,7 @@ defmodule EnumTest do
 
   test "drop every" do
     assert Enum.drop_every([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 2) == [2, 4, 6, 8, 10]
+    assert Enum.drop_every([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 3) == [2, 3, 5, 6, 8, 9]
     assert Enum.drop_every([], 2) == []
     assert Enum.drop_every([1, 2], 2) == [2]
     assert Enum.drop_every([1, 2, 3], 0) == [1, 2, 3]
@@ -468,6 +469,7 @@ defmodule EnumTest do
 
   test "take every" do
     assert Enum.take_every([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 2) == [1, 3, 5, 7, 9]
+    assert Enum.take_every([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 3) == [1, 4, 7, 10]
     assert Enum.take_every([], 2) == []
     assert Enum.take_every([1, 2], 2) == [1]
     assert Enum.take_every([1, 2, 3], 0) == []
diff --git a/lib/elixir/test/elixir/stream_test.exs b/lib/elixir/test/elixir/stream_test.exs
index 262dec146..df4f2442a 100644
--- a/lib/elixir/test/elixir/stream_test.exs
+++ b/lib/elixir/test/elixir/stream_test.exs
@@ -225,6 +225,10 @@ defmodule StreamTest do
            |> Enum.to_list == [2, 4, 6, 8, 10]
 
     assert 1..10
+           |> Stream.drop_every(3)
+           |> Enum.to_list == [2, 3, 5, 6, 8, 9]
+
+    assert 1..10
            |> Stream.drop(2)
            |> Stream.drop_every(2)
            |> Stream.drop(1)
@@ -245,7 +249,7 @@ defmodule StreamTest do
     end
 
     assert_raise FunctionClauseError, fn ->
-      Stream.take_every(1..10, 3.33)
+      Stream.drop_every(1..10, 3.33)
     end
   end
 
@@ -790,6 +794,10 @@ defmodule StreamTest do
            |> Enum.to_list == [1, 3, 5, 7, 9]
 
     assert 1..10
+           |> Stream.take_every(3)
+           |> Enum.to_list == [1, 4, 7, 10]
+
+    assert 1..10
            |> Stream.drop(2)
            |> Stream.take_every(2)
            |> Stream.drop(1)