From 67edb3af571389975cb135aa802487289aa946c8 Mon Sep 17 00:00:00 2001 From: Michael Wu Date: Wed, 18 Jan 2023 16:51:46 -0600 Subject: [PATCH] Add Series.in/2 tests for boolean series --- test/explorer/series_test.exs | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/test/explorer/series_test.exs b/test/explorer/series_test.exs index 2376c4add..e15ebad91 100644 --- a/test/explorer/series_test.exs +++ b/test/explorer/series_test.exs @@ -294,6 +294,13 @@ defmodule Explorer.SeriesTest do end describe "in/2" do + test "with boolean series" do + s1 = Series.from_list([true, false, true]) + s2 = Series.from_list([false, false, false, false]) + + assert s1 |> Series.in(s2) |> Series.to_list() == [false, true, false] + end + test "with integer series" do s1 = Series.from_list([1, 2, 3]) s2 = Series.from_list([1, 0, 3]) @@ -368,6 +375,15 @@ defmodule Explorer.SeriesTest do assert s1 |> Series.in(s2) |> Series.to_list() == [true, false, true] end + test "compare boolean series with an integer series" do + s1 = Series.from_list([true, false, true]) + s2 = Series.from_list([0, 1]) + + assert_raise ArgumentError, fn -> + Series.in(s1, s2) + end + end + test "compare integer series with a float series" do s1 = Series.from_list([1, 2, 3]) s2 = Series.from_list([1.0, 0.5, 3.0])