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