Ejemplo n.º 1
0
    def test_custom_ufunc(self):
        a = np.array([rational(1, 2), rational(1, 3), rational(1, 4)], dtype=rational)
        b = np.array([rational(1, 2), rational(1, 3), rational(1, 4)], dtype=rational)

        result = test_add_rationals(a, b)
        expected = np.array([rational(1), rational(2, 3), rational(1, 2)], dtype=rational)
        assert_equal(result, expected)
Ejemplo n.º 2
0
    def test_custom_ufunc(self):
        a = np.array(
            [rational(1, 2), rational(1, 3),
             rational(1, 4)], dtype=rational)
        b = np.array(
            [rational(1, 2), rational(1, 3),
             rational(1, 4)], dtype=rational)

        result = test_add_rationals(a, b)
        expected = np.array(
            [rational(1), rational(2, 3),
             rational(1, 2)], dtype=rational)
        assert_equal(result, expected)