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_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_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))
