Add a test with multiple element types

Author: PProfizi

tests/test_plotter.py

@@ -231,6 +231,21 @@ def test_field_elemental_nodal_plot_scoped(simple_bar):
     remove_picture(picture)
 
 
+@pytest.mark.skipif(not HAS_PYVISTA, reason="Please install pyvista")
+def test_field_elemental_nodal_plot_multiple_solid_types():
+    model = Model(simple_bar)
+    stress = model.results.element_nodal_forces()
+    fc = stress.outputs.fields_container()
+    f = fc[0]
+    print(f.data.shape)
+    f.plot()
+    picture = 'test_plotter1.png'
+    remove_picture(picture)
+    f.plot(off_screen=True, screenshot=picture)
+    assert os.path.exists(os.path.join(os.getcwd(), picture))
+    remove_picture(picture)
+
+
 @pytest.mark.skipif(not HAS_PYVISTA, reason="Please install pyvista")
 def test_field_solid_plot(allkindofcomplexity):
     model = Model(allkindofcomplexity)