def test_mul(self):
input = gr.SeriesWeld(np.array([1, 2, 3, 4], dtype=np.int32),
gr.WeldInt())
other = gr.SeriesWeld(np.array([1, 2, 3, 4], dtype=np.int32),
gr.WeldInt())
expected_output = [1, 4, 9, 16]
self.assertSequenceEqual(expected_output,
list(input.mul(other).evaluate(False)))
def test_filter(self):
inp = gr.SeriesWeld(np.array(["aaaa", "bbb", "aa", "cccc"], dtype=str),
gr.WeldVec(gr.WeldChar()))
predicates = np.array([True, False, False, True], dtype=bool)
expected_output = ["aaaa", "cccc"]
self.assertSequenceEqual(expected_output,
list(inp.filter(predicates).evaluate(False)))
def test_slice(self):
input = gr.SeriesWeld(
np.array(["aaaaaa", "bbbbbb", "aaaaaaa", "cccccccc"], dtype=str),
gr.WeldVec(gr.WeldChar()))
expected_output = ["aaa", "bbb", "aaa", "ccc"]
self.assertSequenceEqual(expected_output,
list(input.str.slice(1, 3).evaluate(False)))
def test_eq(self):
input = gr.SeriesWeld(
np.array(["aaaa", "bbb", "aa", "cccc"], dtype=str),
gr.WeldVec(gr.WeldChar()))
expected_output = [True, False, False, False]
self.assertSequenceEqual(expected_output,
list((input == "aaaa").evaluate(False)))
def test_eq_and_sum_grouped(self):
inp = gr.SeriesWeld(np.array([4, 3, 2, 1, 3], dtype=np.int32),
gr.WeldInt())
predicates = inp == 3
output = gr.group([predicates, inp.sum()]).evaluate(False)
self.assertSequenceEqual([False, True, False, False, True],
list(output[0]))
self.assertEqual(13, output[1])
def test_eq_and_mask(self):
inp = gr.SeriesWeld(np.array(["aaaa", "bbb", "aa", "cccc"], dtype=str),
gr.WeldVec(gr.WeldChar()))
predicates = inp == "bbb"
expected_output = ["aaaa", "eeee", "aa", "cccc"]
self.assertSequenceEqual(
expected_output,
list(inp.mask(predicates, "eeee").evaluate(False)))
def test_eq_and_mask_grouped_strings(self):
inp = gr.SeriesWeld(np.array(["aaaa", "bbb", "aa", "cccc"], dtype=str),
gr.WeldVec(gr.WeldChar()))
predicates = inp == "bbb"
expected_output = ["aaaa", "eeee", "aa", "cccc"]
output = gr.group([predicates,
inp.mask(predicates, "eeee")]).evaluate(False)
self.assertSequenceEqual([False, True, False, False], list(output[0]))
self.assertSequenceEqual(["aaaa", "eeee", "aa", "cccc"],
list(output[1]))
def test_simpler_eq(self):
input = gr.SeriesWeld(np.array([1, 2, 100, 1, 2, 65], dtype=np.int32),
gr.WeldInt())
expected_output = [False, False, True, False, False, False]
self.assertSequenceEqual(expected_output,
list((input == 100).evaluate(False)))
def test_count(self):
input = gr.SeriesWeld(np.array([1, 2, 3, 4, 5], dtype=np.int32),
gr.WeldInt())
self.assertEqual(5, input.count().evaluate(False))
def test_sum(self):
inp = gr.SeriesWeld(np.array([1, 2, 3, 4, 5], dtype=np.int32),
gr.WeldInt())
self.assertEqual(15, inp.sum().evaluate(False))
def test_unique(self):
input = gr.SeriesWeld(
np.array(["aaa", "bbb", "aaa", "ccc"], dtype=str),
gr.WeldVec(gr.WeldChar()))
self.assertItemsEqual(["aaa", "bbb", "ccc"],
input.unique().evaluate(False))
def test_filter_self(self):
input = gr.SeriesWeld(
np.array(["False", "True", "True", "False"], dtype=str),
gr.WeldVec(gr.WeldChar()))
print input.lower().evaluate()
print input.filter(input.lower() == "true").evaluate()
