Esempio n. 1
0
def test_multi_mesh():
    coordsMulti = np.array([[0.0, 0.0], [1.0, 0.0], [1.0, 1.0], [0.0, 1.0]],
                           dtype=np.float64)

    connect1 = np.array([
        [0, 1, 3],
    ], dtype=np.int64)
    connect2 = np.array([
        [1, 2, 3],
    ], dtype=np.int64)

    data1 = {}
    data2 = {}
    data1["connect1", "test"] = np.array([
        [0.0, 1.0, 3.0],
    ], dtype=np.float64)
    data2["connect2", "test"] = np.array([
        [1.0, 2.0, 3.0],
    ], dtype=np.float64)

    connectList = [connect1, connect2]
    dataList = [data1, data2]

    ds = load_unstructured_mesh(connectList, coordsMulti, dataList)

    sl = SlicePlot(ds, "z", ("connect1", "test"))
    assert sl.data_source.field_data["connect1", "test"].shape == (1, 3)
    sl = SlicePlot(ds, "z", ("connect2", "test"))
    assert sl.data_source.field_data["connect2", "test"].shape == (1, 3)
    sl = SlicePlot(ds, "z", ("all", "test"))
    assert sl.data_source.field_data["all", "test"].shape == (2, 3)
    sl.annotate_mesh_lines()
Esempio n. 2
0
def test_multi_mesh():
    coordsMulti = np.array([[0.0, 0.0], [1.0, 0.0], [1.0, 1.0], [0.0, 1.0]],
                           dtype=np.float64)

    connect1 = np.array([
        [0, 1, 3],
    ], dtype=np.int64)
    connect2 = np.array([
        [1, 2, 3],
    ], dtype=np.int64)

    data1 = {}
    data2 = {}
    data1['connect1', 'test'] = np.array([
        [0.0, 1.0, 3.0],
    ], dtype=np.float64)
    data2['connect2', 'test'] = np.array([
        [1.0, 2.0, 3.0],
    ], dtype=np.float64)

    connectList = [connect1, connect2]
    dataList = [data1, data2]

    ds = load_unstructured_mesh(connectList, coordsMulti, dataList)

    sl = SlicePlot(ds, 'z', ('connect1', 'test'))
    sl = SlicePlot(ds, 'z', ('connect2', 'test'))
    sl = SlicePlot(ds, 'z', ('all', 'test'))
    sl.annotate_mesh_lines()
Esempio n. 3
0
def test_multi_field():
    coords = np.array([[0.0, 0.0], [1.0, 0.0], [1.0, 1.0], [0.0, 1.0]],
                      dtype=np.float64)

    connect = np.array([[0, 1, 3], [1, 2, 3]], dtype=np.int64)

    data = {}
    data["connect1", "test"] = np.array([[0.0, 1.0, 3.0], [1.0, 2.0, 3.0]],
                                        dtype=np.float64)
    data["connect1",
         "testAgain"] = np.array([[0.0, 1.0, 3.0], [1.0, 2.0, 3.0]],
                                 dtype=np.float64)

    ds = load_unstructured_mesh(connect, coords, data)

    sl = SlicePlot(ds, "z", "test")
    sl.annotate_mesh_lines()

    sl = SlicePlot(ds, "z", "testAgain")
    sl.annotate_mesh_lines()