yaq-python/tests/ndarray/test_ndarray.py at de1112ba2bbdacc813e8043d7f91904dac8114e2 · yaq-project/yaq-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import pathlib
import subprocess
import sys
import time

import numpy as np

import yaqc
from yaqd_core import testing

pyfile = pathlib.Path(__file__).parent / "NdarrayTestDaemon.py"
config = pyfile.with_suffix(".toml")


@testing.run_daemon_from_file(pyfile, config)
def test_subtract_int():
    s = yaqc.Client(38001)
    result = s.subtract(minuend=np.arange(8), subtrahend=-1 * np.arange(8))
    np.testing.assert_equal(result, np.arange(8) * 2)


@testing.run_daemon_from_file(pyfile, config)
def test_sum_float():
    s = yaqc.Client(38001)
    result = s.sum(np.linspace(0, 10))
    np.testing.assert_equal(result, np.sum(np.linspace(0, 10)))


@testing.run_daemon_from_file(pyfile, config)
def test_shape():
    s = yaqc.Client(38001)
    result = s.shape()
    assert result.shape == (2, 5)
    np.testing.assert_equal(result, np.linspace(0, 1, 10).reshape(2, 5))


@testing.run_daemon_from_file(pyfile, config)
def test_union():
    s = yaqc.Client(38001)
    result = s.union()
    np.testing.assert_equal(result, np.linspace(0, 1, 10).reshape(2, 5))


@testing.run_daemon_from_file(pyfile, config)
def test_map_union():
    s = yaqc.Client(38001)
    result = s.map_union()
    assert tuple(result.keys()) == ("simple", "complicated")
    # assert result["complicated"].shape == (2,5)