pytecplot 1.2.1__zip → 1.3.1__zip

{pytecplot-1.2.1 → pytecplot-1.3.1r110473}/test/data/test_query.py

@@ -708,7 +708,6 @@ class TestProbeOnSurface(unittest.TestCase):
             self.assertEqual(p, IJKPlanes.Unused)
             self.assertEqual(z, ds.zone(0))
 
-
     def test_poly_triangle(self):
         tp.new_layout()
         fr = tp.active_frame()
@@ -914,6 +913,68 @@ class TestProbeOnSurface(unittest.TestCase):
             self.assertEqual(p, IJKPlanes.Unused)
             self.assertEqual(z, ds.zone(0))
 
+    @skip_if_sdk_version_before(2020, 0)
+    def test_skew_fe_quad(self):
+        tp.new_layout()
+        fr = tp.active_frame()
+        ds = fr.create_dataset('D', ['x', 'y', 'z', 's', 't'])
+        zn = ds.add_fe_zone(ZoneType.FEQuad, 'Z', 4, 1)
+        fr.plot_type = PlotType.Cartesian3D
+
+        nodes = [[0, 0, 0. ],
+                 [1, 0, 0.001],
+                 [1, 1, 0. ],
+                 [0, 1, 0.001]]
+        conn = [[0, 1, 2, 3]]
+
+        zn.values('x')[:] = [n[0] for n in nodes]
+        zn.values('y')[:] = [n[1] for n in nodes]
+        zn.values('z')[:] = [n[2] for n in nodes]
+        zn.values('s')[:] = [0,1,2,3]
+        zn.values('t')[:] = [0,0,2,2]
+
+        zn.nodemap[:] = conn
+
+        points = np.array([
+            [.6, .4, -0.2],
+            [.6, .4,  0. ],
+            [.6, .4,  0.2],
+            ])
+
+        ### Test 1: nominal tolerance considers this quad to be invalid
+        ###         and will return the nearest node.
+        expected = [
+            [ 1, 0, 0.001, 1, 0],
+            [ 1, 0, 0.001, 1, 0],
+            [ 1, 0, 0.001, 1, 0],]
+
+        v,cells,planes,zones = tp.data.query.probe_on_surface(points.T)
+        vv = np.array(v).reshape((-1, len(points)))
+
+        for ex, v, c, p, z in zip(expected, vv.T, cells, planes, zones):
+            for i in range(5):
+                self.assertAlmostEqual(v[i], ex[i])
+            self.assertEqual(c, 0)
+            self.assertEqual(p, IJKPlanes.Unused)
+            self.assertEqual(z, ds.zone(0))
+
+        ### Test 2: large tolerance allows the calculation to proceed with
+        ###         the skew quad cell treating it as if it were flat.
+        expected = [
+            [ 0.6, 0.4, -0.0002, 1.4, 0.8],
+            [ 0.6, 0.4,  0.0002, 1.4, 0.8],
+            [ 0.6, 0.4,  0.0002, 1.4, 0.8],]
+
+        v,cells,planes,zones = tp.data.query.probe_on_surface(points.T, tolerance=0.01)
+        vv = np.array(v).reshape((-1, len(points)))
+
+        for ex, v, c, p, z in zip(expected, vv.T, cells, planes, zones):
+            for i in range(5):
+                self.assertAlmostEqual(v[i], ex[i], delta=0.001)
+            self.assertEqual(c, 0)
+            self.assertEqual(p, IJKPlanes.Unused)
+            self.assertEqual(z, ds.zone(0))
+
 
 if __name__ == '__main__':
     from .. import main

pytecplot-1.3.1r110473/test/export/test_time_animation.py

@@ -0,0 +1,185 @@
+import imghdr
+import os
+import tempfile
+import unittest
+from unittest.mock import *
+
+import tecplot as tp
+from tecplot.exception import *
+from tecplot.constant import *
+
+from ..sample_data import sample_data
+from .. import (patch_tecutil, closed_tempfile, skip_if_sdk_version_before,
+                mocked_sdk_version)
+from .test_animation import isvideo
+
+
+class TestTimeAnimation(unittest.TestCase):
+    _OPTS = dict(width=32, supersample=1)
+
+    def setUp(self):
+        tp.new_layout()
+        self.filename, _ = sample_data('3x3x3_3zones_time')
+        tp.active_frame().plot(PlotType.Cartesian3D).activate()
+
+    def tearDown(self):
+        tp.new_layout()
+        os.remove(self.filename)
+
+    def test_animation_avi(self):
+        opts = TestTimeAnimation._OPTS.copy()
+        opts.update(timestep_step = None)
+        with closed_tempfile('.avi') as fname:
+            tp.export.save_time_animation_avi(fname, **opts)
+            self.assertTrue(isvideo(self, fname))
+
+    def test_mpeg4_failure_on_out_of_date_engine(self):
+        with mocked_sdk_version(2017, 2):
+            with closed_tempfile('.avi') as fname:
+                tp.export.save_time_animation_avi(fname, **TestTimeAnimation._OPTS)
+                self.assertTrue(isvideo(self, fname))
+            with self.assertRaises(TecplotOutOfDateEngineError):
+                with closed_tempfile('.mp4') as fname:
+                    tp.export.save_time_animation(fname, **TestTimeAnimation._OPTS)
+
+    @skip_if_sdk_version_before(2017, 3)
+    def test_animation_mpeg4(self):
+        opts = TestTimeAnimation._OPTS.copy()
+        opts.update(
+            start_time = 1,
+            end_time = 2,
+            timestep_step = 1,
+        )
+        with closed_tempfile('.mp4') as fname:
+            tp.export.save_time_animation_mpeg4(fname, **opts)
+            self.assertTrue(isvideo(self, fname))
+
+    @skip_if_sdk_version_before(2017, 3)
+    def test_animation_mpeg4_byext(self):
+        with closed_tempfile('.mp4') as fname:
+            tp.export.save_time_animation(fname, **TestTimeAnimation._OPTS)
+            self.assertTrue(isvideo(self, fname))
+
+    @skip_if_sdk_version_before(2017, 3)
+    def test_animation_wmv(self):
+        with closed_tempfile('.wmv') as fname:
+            tp.export.save_time_animation_wmv(fname, **TestTimeAnimation._OPTS)
+            self.assertTrue(isvideo(self, fname))
+
+    def test_animation_raster_metafile(self):
+        with closed_tempfile('.rm') as fname:
+            tp.export.save_time_animation_raster_metafile(fname, **TestTimeAnimation._OPTS)
+            # no good way to determine video format for
+            # raster metafiles so just check the file size
+            with open(fname, 'rb') as fin:
+                self.assertGreater(len(fin.read()), 800)
+
+    def test_animation_flash(self):
+        with closed_tempfile('.flv') as fname:
+            opts = TestTimeAnimation._OPTS.copy()
+            opts.update(
+                compression = FlashCompressionType.SmallestSize,
+                image_type = FlashImageType.Color256
+            )
+            tp.export.save_time_animation_flash(fname, **opts)
+            self.assertTrue(isvideo(self, fname))
+
+    @skip_if_sdk_version_before(2017, 3)
+    def test_failures(self):
+        with closed_tempfile('.mp4') as fname:
+            with patch_tecutil('AnimateTimeX', return_value=False):
+                with self.assertRaises(TecplotSystemError):
+                    tp.export.save_time_animation_mpeg4(fname)
+
+    @skip_if_sdk_version_before(2018, 2)
+    def test_animation_bmp(self):
+        opts = TestTimeAnimation._OPTS.copy()
+        with tempfile.TemporaryDirectory() as tmpdir:
+            cwd = os.getcwd()
+            try:
+                os.chdir(tmpdir)
+                tp.export.save_time_animation_bmp('img.bmp', **opts)
+                self.assertFalse(os.path.exists('img.bmp'))
+                self.assertTrue(os.path.exists('img_000001.bmp'))
+                self.assertTrue(os.path.exists('img_000002.bmp'))
+                self.assertTrue(os.path.exists('img_000003.bmp'))
+                self.assertFalse(os.path.exists('img_000004.bmp'))
+                self.assertEqual(imghdr.what('img_000001.bmp'), 'bmp')
+            finally:
+                os.chdir(cwd)
+
+    @skip_if_sdk_version_before(2018, 2)
+    def test_animation_jpeg(self):
+        opts = TestTimeAnimation._OPTS.copy()
+        with tempfile.TemporaryDirectory() as tmpdir:
+            cwd = os.getcwd()
+            try:
+                os.chdir(tmpdir)
+                tp.export.save_time_animation_jpeg('img.jpeg', **opts)
+                self.assertFalse(os.path.exists('img.jpeg'))
+                self.assertTrue(os.path.exists('img_000001.jpeg'))
+                self.assertTrue(os.path.exists('img_000002.jpeg'))
+                self.assertTrue(os.path.exists('img_000003.jpeg'))
+                self.assertFalse(os.path.exists('img_000004.jpeg'))
+                self.assertEqual(imghdr.what('img_000001.jpeg'), 'jpeg')
+            finally:
+                os.chdir(cwd)
+
+    @skip_if_sdk_version_before(2018, 2)
+    def test_animation_png(self):
+        opts = TestTimeAnimation._OPTS.copy()
+        with tempfile.TemporaryDirectory() as tmpdir:
+            cwd = os.getcwd()
+            try:
+                os.chdir(tmpdir)
+                tp.export.save_time_animation_png('img.png', **opts)
+                self.assertFalse(os.path.exists('img.png'))
+                self.assertTrue(os.path.exists('img_000001.png'))
+                self.assertTrue(os.path.exists('img_000002.png'))
+                self.assertTrue(os.path.exists('img_000003.png'))
+                self.assertFalse(os.path.exists('img_000004.png'))
+                self.assertEqual(imghdr.what('img_000001.png'), 'png')
+            finally:
+                os.chdir(cwd)
+
+    @skip_if_sdk_version_before(2018, 2)
+    def test_animation_tiff(self):
+        opts = TestTimeAnimation._OPTS.copy()
+        with tempfile.TemporaryDirectory() as tmpdir:
+            cwd = os.getcwd()
+            try:
+                os.chdir(tmpdir)
+                tp.export.save_time_animation_tiff('img.tiff', **opts)
+                self.assertFalse(os.path.exists('img.tiff'))
+                self.assertTrue(os.path.exists('img_000001.tiff'))
+                self.assertTrue(os.path.exists('img_000002.tiff'))
+                self.assertTrue(os.path.exists('img_000003.tiff'))
+                self.assertFalse(os.path.exists('img_000004.tiff'))
+                self.assertEqual(imghdr.what('img_000001.tiff'), 'tiff')
+            finally:
+                os.chdir(cwd)
+
+    def test_images_failure_on_out_of_date_engine(self):
+        if __debug__:
+            opts = TestTimeAnimation._OPTS.copy()
+            with mocked_sdk_version(2018, 1):
+                with tempfile.TemporaryDirectory() as tmpdir:
+                    cwd = os.getcwd()
+                    try:
+                        os.chdir(tmpdir)
+                        with self.assertRaises(TecplotOutOfDateEngineError):
+                            tp.export.save_time_animation_bmp('img.bmp', **opts)
+                        with self.assertRaises(TecplotOutOfDateEngineError):
+                            tp.export.save_time_animation_jpeg('img.jpeg', **opts)
+                        with self.assertRaises(TecplotOutOfDateEngineError):
+                            tp.export.save_time_animation_png('img.png', **opts)
+                        with self.assertRaises(TecplotOutOfDateEngineError):
+                            tp.export.save_time_animation_tiff('img.tiff', **opts)
+                    finally:
+                        os.chdir(cwd)
+
+
+if __name__ == '__main__':
+    from .. import main
+    main()
+

{pytecplot-1.2.1 → pytecplot-1.3.1r110473}/test/plot/test_axis.py

@@ -2,6 +2,7 @@ import numpy as np
 import os
 import random
 import sys
+import tempfile
 import unittest
 
 from os import path
@@ -315,6 +316,26 @@ class TestFieldAxis(object):
     def test_fit_range_with_blanking(self):
         self.axes[0].fit_range(True)
 
+    def test_minmax_with_new_axis_variable(self):
+        fr = tp.active_frame()
+        plot = fr.plot()
+        dataset = fr.dataset
+        tp.data.operate.execute_equation('{a}={X}')
+        plot.axes.x_axis.variable = dataset.variable('a')
+        plot.axes.x_axis.max = 2.0
+        self.assertAlmostEqual(plot.axes.x_axis.max, 2.0)
+        plot.axes.x_axis.max = 4.0
+        self.assertAlmostEqual(plot.axes.x_axis.max, 4.0)
+
+        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as ftmp:
+            ftmp.close()
+            try:
+                tp.export.save_png(ftmp.name, 10, supersample=1)
+            finally:
+                os.remove(ftmp.name)
+
+        self.assertAlmostEqual(plot.axes.x_axis.max, 4.0)
+
 
 class TestCartesian2DFieldAxis(unittest.TestCase, TestCartesian2DAxis,
                                TestFieldAxis, TestAxis):

{pytecplot-1.2.1 → pytecplot-1.3.1r110473}/test/session/test_session.py

@@ -1,6 +1,12 @@
 from __future__ import unicode_literals
 
-import datetime, os, unittest, warnings
+import atexit
+import datetime
+import os
+import platform
+import sys
+import unittest
+import warnings
 
 from contextlib import contextmanager
 from datetime import date
@@ -197,6 +203,34 @@ class TestSession(unittest.TestCase):
                         self.assertGreater(disconn.call_count, 0)
                         self.assertGreater(quit.call_count, 0)
 
+    def test_atexit(self):
+        try:
+            connected = tp.tecutil._tecutil_connector.connected
+            if connected:
+                if sys.version_info < (3,):
+                    handlers_to_delete = []
+                    for i in range(0, len(atexit._exithandlers)):
+                        if (
+                            len(atexit._exithandlers[i]) and
+                            atexit._exithandlers[i][0] == tp.tecutil._tecutil_connector.client.disconnect
+                        ):
+                            handlers_to_delete.append(i)
+                    for i in reversed(sorted(handlers_to_delete)):
+                        del atexit._exithandlers[i]
+                else:
+                    atexit.unregister(tp.tecutil._tecutil_connector.client.disconnect)
+            with patch.object(tp.tecutil._tecutil_connector, 'stop') as stop:
+                with patch.object(tp.tecutil._tecutil_connector, 'release_license') as release_license:
+                    atexit._run_exitfuncs()
+                    if platform.system() == 'Darwin':
+                        self.assertEqual(stop.call_count, 0)
+                        self.assertEqual(release_license.call_count, 1)
+                    else:
+                        self.assertEqual(stop.call_count, 1)
+        finally:
+            if connected:
+                atexit.register(tp.tecutil._tecutil_connector.client.disconnect)
+
 
 if __name__ == '__main__':
     from .. import main

{pytecplot-1.2.1 → pytecplot-1.3.1r110473}/tox.ini

@@ -1,12 +1,12 @@
 [tox]
-envlist = py27, py{36,37,38}, examples, doc
+envlist = py27, py{37,38,39}, examples, doc
 
 [testenv]
 usedevelop = True
 extras = test
 args_are_paths = False
 passenv = DISPLAY USERNAME USER HOME HOMEPATH HOMEDRIVE DYLD_LIBRARY_PATH LD_LIBRARY_PATH RLM_LICENSE TECSDKHOME LD_PRELOAD
-commands = python test/run-tests.py {posargs:--no-coverage test --failfast} --buffer
+commands = python test/run-tests.py {posargs:--no-coverage test --info --failfast} --buffer
 
 [testenv:py27]
 deps =
@@ -28,4 +28,3 @@ changedir = docs
 commands =
     make clean
     python build-html
-