Add "unit tests" for diff tool to increase coverage

asdf-format · Aug 4, 2017 · 60e762b · 60e762b
1 parent 30e12c9
commit 60e762b
Show file tree

Hide file tree

Showing 9 changed files with 345 additions and 0 deletions.
diff --git a/asdf/commands/tests/data/block0.asdf b/asdf/commands/tests/data/block0.asdf
diff --git a/asdf/commands/tests/data/block1.asdf b/asdf/commands/tests/data/block1.asdf
diff --git a/asdf/commands/tests/data/blocks.diff b/asdf/commands/tests/data/blocks.diff
@@ -0,0 +1,13 @@
+tree:[0m
+  foobar:[0m
+    bizbaz:[0m
+[32m>     green[0m
+[31m<     red[0m
+    datatype:[0m
+[32m>     uint64[0m
+[31m<     float64[0m
+    shape:[0m
+      -[0m
+[32m>       9000[0m
+[31m<       10000[0m
+    [31mndarrays differ by shape, datatype and contents[0m
diff --git a/asdf/commands/tests/data/frames.diff b/asdf/commands/tests/data/frames.diff
@@ -0,0 +1,46 @@
+tree:[0m
+  asdf_library:[0m
+    version:[0m
+[32m>     1.2.2.dev858[0m
+[31m<     1.2.2.dev846[0m
+[32m> frames:[0m
+[32m>   - reference_frame:[0m
+[32m>       galcen_coord:[0m
+[32m>         dec:[0m
+[32m>           unit:[0m
+[32m>             deg[0m
+[32m>           value:[0m
+[32m>             -28.936175[0m
+[32m>         ra:[0m
+[32m>           unit:[0m
+[32m>             deg[0m
+[32m>           value:[0m
+[32m>             266.4051[0m
+[32m>           wrap_angle:[0m
+[32m>             unit:[0m
+[32m>               deg[0m
+[32m>             value:[0m
+[32m>               360.0[0m
+[32m>       galcen_v_sun:[0m
+[32m>         - unit:[0m
+[32m>             km s-1[0m
+[32m>           value:[0m
+[32m>             11.1[0m
+[32m>         - unit:[0m
+[32m>             km s-1[0m
+[32m>           value:[0m
+[32m>             232.24[0m
+[32m>         - unit:[0m
+[32m>             km s-1[0m
+[32m>           value:[0m
+[32m>             7.25[0m
+[31m<       galcen_dec:[0m
+[31m<         unit:[0m
+[31m<           rad[0m
+[31m<         value:[0m
+[31m<           1.0[0m
+[31m<       galcen_ra:[0m
+[31m<         unit:[0m
+[31m<           deg[0m
+[31m<         value:[0m
+[31m<           45.0[0m
diff --git a/asdf/commands/tests/data/frames0.asdf b/asdf/commands/tests/data/frames0.asdf
@@ -0,0 +1,97 @@
+#ASDF 1.0.0
+#ASDF_STANDARD 1.1.0
+%YAML 1.1
+%TAG ! tag:stsci.edu:asdf/
+--- !core/asdf-1.0.0
+asdf_library: !core/software-1.0.0 {author: Space Telescope Science Institute, homepage: 'http://github.com/spacetelescope/asdf',
+  name: asdf, version: 1.2.2.dev858}
+frames:
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {type: ICRS}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {equinox: !time/time-1.0.0 '2010-01-01 00:00:00.000', type: FK5}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {equinox: !time/time-1.0.0 '2010-01-01 00:00:00.000', obstime: !time/time-1.0.0 '2015-01-01
+      00:00:00.000', type: FK4}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {equinox: !time/time-1.0.0 '2010-01-01 00:00:00.000', obstime: !time/time-1.0.0 '2015-01-01
+      00:00:00.000', type: FK4_noeterms}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {type: galactic}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [x, y, z]
+  axes_order: [0, 1, 2]
+  name: CelestialFrame
+  reference_frame:
+    galcen_coord: !wcs/icrs_coord-1.1.0
+      dec: {value: -28.936175}
+      ra:
+        value: 266.4051
+        wrap_angle: !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 deg, value: 360.0}
+    galcen_distance: !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m, value: 5.0}
+    galcen_v_sun:
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 km s-1, value: 11.1}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 km s-1, value: 232.24}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 km s-1, value: 7.25}
+    roll: !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 deg, value: 3.0}
+    type: galactocentric
+    z_sun: !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 pc, value: 3.0}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame:
+    obsgeoloc:
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m, value: 3.0856775814671916e+16}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m, value: 9.257032744401574e+16}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m, value: 6.1713551629343834e+19}
+    obsgeovel:
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m s-1, value: 2.0}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m s-1, value: 1.0}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m s-1, value: 8.0}
+    obstime: !time/time-1.0.0 2010-01-01 00:00:00.000
+    type: GCRS
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {obstime: !time/time-1.0.0 '2010-01-01 00:00:00.000', type: CIRS}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [x, y, z]
+  axes_order: [0, 1, 2]
+  name: CelestialFrame
+  reference_frame: {obstime: !time/time-1.0.0 '2022-01-03 00:00:00.000', type: ITRS}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame:
+    equinox: !time/time-1.0.0 J2000.000
+    obsgeoloc:
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m, value: 3.0856775814671916e+16}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m, value: 9.257032744401574e+16}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m, value: 6.1713551629343834e+19}
+    obsgeovel:
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m s-1, value: 2.0}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m s-1, value: 1.0}
+    - !unit/quantity-1.1.0 {unit: !unit/unit-1.0.0 m s-1, value: 8.0}
+    obstime: !time/time-1.0.0 2010-01-01 00:00:00.000
+    type: precessed_geocentric
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+...
diff --git a/asdf/commands/tests/data/frames1.asdf b/asdf/commands/tests/data/frames1.asdf
@@ -0,0 +1,124 @@
+#ASDF 1.0.0
+#ASDF_STANDARD 1.1.0
+%YAML 1.1
+%TAG ! tag:stsci.edu:asdf/
+--- !core/asdf-1.0.0
+asdf_library: !core/software-1.0.0 {author: Space Telescope Science Institute, homepage: 'http://github.com/spacetelescope/asdf',
+  name: asdf, version: 1.2.2.dev846}
+frames:
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat, blurg]
+  name: CelestialFrame
+  reference_frame: {type: ICRS}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {equinox: !time/time-1.0.0 '2010-01-01 00:00:00.000', type: FK5}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {equinox: !time/time-1.0.0 '2010-01-01 00:00:00.000', obstime: !time/time-1.0.0 '2015-01-01
+      00:00:00.000', type: FK4}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {equinox: !time/time-1.0.0 '2010-01-01 00:00:00.000', obstime: !time/time-1.0.0 '2015-01-01
+      00:00:00.000', type: FK4_noeterms}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {type: galactic}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [x, y, z]
+  axes_order: [0, 1, 2]
+  name: CelestialFrame
+  reference_frame:
+    galcen_dec: !unit/quantity-1.1.0
+      unit: rad
+      value: 1.0
+    galcen_distance: !unit/quantity-1.1.0
+      unit: m
+      value: 5.0
+    galcen_ra: !unit/quantity-1.1.0
+      unit: deg
+      value: 45.0
+    roll: !unit/quantity-1.1.0
+      unit: deg
+      value: 3.0
+    type: galactocentric
+    z_sun: !unit/quantity-1.1.0
+      unit: pc
+      value: 3.0
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame:
+    obsgeoloc:
+    - !unit/quantity-1.1.0
+      unit: m
+      value: 3.0856775814671916e+16
+    - !unit/quantity-1.1.0
+      unit: m
+      value: 9.257032744401574e+16
+    - !unit/quantity-1.1.0
+      unit: m
+      value: 6.1713551629343834e+19
+    obsgeovel:
+    - !unit/quantity-1.1.0
+      unit: m s-1
+      value: 2.0
+    - !unit/quantity-1.1.0
+      unit: m s-1
+      value: 1.0
+    - !unit/quantity-1.1.0
+      unit: m s-1
+      value: 8.0
+    obstime: !time/time-1.0.0 2010-01-01 00:00:00.000
+    type: GCRS
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame: {obstime: !time/time-1.0.0 '2010-01-01 00:00:00.000', type: CIRS}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [x, y, z]
+  axes_order: [0, 1, 2]
+  name: CelestialFrame
+  reference_frame: {obstime: !time/time-1.0.0 '2022-01-03 00:00:00.000', type: ITRS}
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+- !wcs/celestial_frame-1.1.0
+  axes_names: [lon, lat]
+  name: CelestialFrame
+  reference_frame:
+    equinox: !time/time-1.0.0 J2000.000
+    obsgeoloc:
+    - !unit/quantity-1.1.0
+      unit: m
+      value: 3.0856775814671916e+16
+    - !unit/quantity-1.1.0
+      unit: m
+      value: 9.257032744401574e+16
+    - !unit/quantity-1.1.0
+      unit: m
+      value: 6.1713551629343834e+19
+    obsgeovel:
+    - !unit/quantity-1.1.0
+      unit: m s-1
+      value: 2.0
+    - !unit/quantity-1.1.0
+      unit: m s-1
+      value: 1.0
+    - !unit/quantity-1.1.0
+      unit: m s-1
+      value: 8.0
+    obstime: !time/time-1.0.0 2010-01-01 00:00:00.000
+    type: precessed_geocentric
+  unit: [!unit/unit-1.0.0 deg, !unit/unit-1.0.0 deg]
+...
diff --git a/asdf/commands/tests/data/frames_minimal.diff b/asdf/commands/tests/data/frames_minimal.diff
@@ -0,0 +1,12 @@
+tree:[0m
+  asdf_library:[0m
+    version:[0m
+[32m>     1.2.2.dev858[0m
+[31m<     1.2.2.dev846[0m
+  frames:[0m
+    -[0m
+      reference_frame:[0m
+[32m>       galcen_coord[0m
+[32m>       galcen_v_sun[0m
+[31m<       galcen_dec[0m
+[31m<       galcen_ra[0m
diff --git a/asdf/commands/tests/setup_package.py b/asdf/commands/tests/setup_package.py
@@ -0,0 +1,9 @@
+# Licensed under a 3-clause BSD style license - see LICENSE.rst
+# -*- coding: utf-8 -*-
+
+from __future__ import absolute_import, division, unicode_literals, print_function
+
+
+def get_package_data():  # pragma: no cover
+    return {
+        str(_PACKAGE_NAME_ + '.commands.tests'): ['data/*.asdf', 'data/*.diff']}
diff --git a/asdf/commands/tests/test_diff.py b/asdf/commands/tests/test_diff.py
@@ -0,0 +1,44 @@
+# Licensed under a 3-clause BSD style license - see LICENSE.rst
+# -*- coding: utf-8 -*-
+
+from __future__ import absolute_import, division, unicode_literals, print_function
+
+import os
+import io
+
+import numpy as np
+import pytest
+
+from ... import AsdfFile
+from .. import diff
+
+
+TEST_DATA_PATH = os.path.join(os.path.dirname(__file__), 'data')
+
+
+def _assert_diffs_equal(filenames, result_file, minimal=False):
+    iostream = io.StringIO()
+
+    file_paths = ["{}/{}".format(TEST_DATA_PATH, name) for name in filenames]
+    diff(file_paths, minimal=minimal, iostream=iostream)
+    iostream.seek(0)
+
+    result_path = "{}/{}".format(TEST_DATA_PATH, result_file)
+    with open(result_path, 'r') as handle:
+        assert handle.read() == iostream.read()
+
+def test_diff():
+    filenames = ['frames0.asdf', 'frames1.asdf']
+    result_file = 'frames.diff'
+    _assert_diffs_equal(filenames, result_file, minimal=False)
+
+def test_diff_minimal():
+    filenames = ['frames0.asdf', 'frames1.asdf']
+    result_file = 'frames_minimal.diff'
+    _assert_diffs_equal(filenames, result_file, minimal=True)
+
+def test_diff_block():
+    filenames = ['block0.asdf', 'block1.asdf']
+    result_file = 'blocks.diff'
+
+    _assert_diffs_equal(filenames, result_file, minimal=False)