Merge pull request #314 from nbirkbeck/unit-tests3

Adding unit tests for spherical metadata injection.

Author: jamiesonpepper

data/testsrc_320x240_h264.mp4

@@ -34,7 +34,7 @@ def console(contents):
  print(contents)
 
 
-def main():
+def main(main_args):
  """Main function for printing and injecting spatial media metadata."""
 
  parser = argparse.ArgumentParser(
@@ -90,7 +90,7 @@ def main():
  "ordering and SN3D normalization")
  parser.add_argument("file", nargs="+", help="input/output files")
 
- args = parser.parse_args()
+ args = parser.parse_args(main_args)
 
  if args.inject:
  if len(args.file) != 2:
@@ -141,4 +141,4 @@ def main():
 
 
 if __name__ == "__main__":
- main()
+ main(sys.argv[1:])

data/testsrc_320x240_vp9.mp4

@@ -0,0 +1,127 @@
+"""
+Unit tests for the spatialmedia CLI tool.
+
+# Test inputs are generated with
+ffmpeg -y -f lavfi -i testsrc -vf scale=320:240 -vcodec libx264 -t 0.05 data/testsrc_320x240_h264.mp4
+ffmpeg -y -f lavfi -i testsrc -vf scale=320:240 -vcodec libvpx-vp9 -t 0.05 data/testsrc_320x240_vp9.mp4
+ffmpeg -y -f lavfi -i testsrc -vf scale=32:24 -vcodec prores -t 0.05 data/testsrc_32x24_prores.mov
+
+"""
+import unittest
+import os
+
+from spatialmedia.__main__ import main
+from spatialmedia import metadata_utils
+
+_OUTPUT_DIR = 'test_output'
+
+def append_contents(contents):
+ def append(x):
+ contents = contents + x
+ return append
+
+class TestAdd(unittest.TestCase):
+
+ def inject_metadata(self, args):
+ self.assertIsNone(main(args))
+
+ contents = []
+ metadata_utils.parse_metadata(args[-1],
+ lambda x: contents.append(x))
+ try:
+ os.remove(args[-1])
+ except:
+ return ''
+ return '\n'.join(contents[2:])
+
+ def test_inject_v1_equirect_mono(self):
+ contents = self.inject_metadata(['-i',
+ '--projection', 'equirectangular',
+ 'data/testsrc_320x240_h264.mp4',
+ f'{_OUTPUT_DIR}/equirect_mono_v1.mp4'])
+
+ self.assertFalse(contents.find('SV3D') >= 0)
+ self.assertFalse(contents.find('PRHD') >= 0)
+ self.assertFalse(contents.find('EQUI') >= 0)
+ self.assertFalse(contents.find('ST3D') >= 0)
+ self.assertTrue(contents.find("ProjectionType = equirectangular") > 0)
+
+ def test_inject_v1_rectangular_left_right(self):
+ contents = self.inject_metadata(['-i',
+ '--stereo', 'left-right',
+ '--projection', 'none',
+ 'data/testsrc_320x240_h264.mp4',
+ f'{_OUTPUT_DIR}/rectangular_left_right_v1.mp4'])
+
+ self.assertFalse(contents.find('SV3D') >= 0)
+ self.assertFalse(contents.find('PRHD') >= 0)
+ self.assertFalse(contents.find('EQUI') >= 0)
+ self.assertFalse(contents.find('ST3D') >= 0)
+ self.assertFalse(contents.find("ProjectionType = equirectangular") > 0)
+ self.assertTrue(contents.find("ProjectionType = rectangular") > 0)
+ self.assertTrue(contents.find("Spherical = false") > 0)
+
+ def test_inject_v2_equirect_mono(self):
+ contents = self.inject_metadata(['-i',
+ '--v2', '--projection', 'equirectangular',
+ 'data/testsrc_320x240_h264.mp4',
+ f'{_OUTPUT_DIR}/equirect_mono.mp4'])
+ self.assertTrue(contents.find('SV3D') >= 0)
+ self.assertTrue(contents.find('PRHD') >= 0)
+ self.assertTrue(contents.find('EQUI') >= 0)
+ self.assertFalse(contents.find('ST3D') >= 0)
+
+ def test_inject_v2_equirect_mono_vp9(self):
+ contents = self.inject_metadata(['-i',
+ '--v2', '--projection', 'equirectangular',
+ 'data/testsrc_320x240_vp9.mp4',
+ f'{_OUTPUT_DIR}/equirect_mono_vp9.mp4'])
+
+ self.assertTrue(contents.find('SV3D') >= 0)
+ self.assertTrue(contents.find('PRHD') >= 0)
+ self.assertTrue(contents.find('EQUI') >= 0)
+ self.assertFalse(contents.find('ST3D') >= 0)
+
+ def test_inject_v2_equirect_mono_prores(self):
+ contents = self.inject_metadata(['-i',
+ '--v2', '--projection', 'equirectangular',
+ 'data/testsrc_32x24_prores.mov',
+ f'{_OUTPUT_DIR}/equirect_mono_prores.mov'])
+ self.assertTrue(contents.find('SV3D') >= 0)
+ self.assertTrue(contents.find('PRHD') >= 0)
+ self.assertTrue(contents.find('EQUI') >= 0)
+ self.assertFalse(contents.find('ST3D') >= 0)
+
+
+ def test_inject_v2_rectangular_left_right(self):
+ contents = self.inject_metadata(['-i',
+ '--v2', '--stereo', 'left-right',
+ '--projection', 'none',
+ 'data/testsrc_320x240_h264.mp4',
+ f'{_OUTPUT_DIR}/rectangular_left_right.mp4'])
+ self.assertFalse(contents.find('SV3D') >= 0)
+ self.assertFalse(contents.find('PRHD') >= 0)
+ self.assertFalse(contents.find('EQUI') >= 0)
+ self.assertTrue(contents.find('ST3D') >= 0)
+ self.assertTrue(contents.find('Stereo Mode: 2') >= 0)
+
+ def test_inject_v2_rectangular_top_bottom(self):
+ contents = self.inject_metadata(['-i',
+ '--v2', '--stereo', 'top-bottom',
+ '--projection', 'none',
+ 'data/testsrc_320x240_h264.mp4',
+ f'{_OUTPUT_DIR}/rectangular_top_bottom.mp4'])
+
+ self.assertFalse(contents.find('SV3D') >= 0)
+ self.assertFalse(contents.find('PRHD') >= 0)
+ self.assertFalse(contents.find('EQUI') >= 0)
+ self.assertTrue(contents.find('ST3D') >= 0)
+ self.assertTrue(contents.find('Stereo Mode: 1') >= 0)
+
+
+if __name__ == '__main__':
+ try:
+ os.mkdir('test_output')
+ except:
+ pass
+ unittest.main()