def test_extract_artifact_as_visualization(self): artifact = Artifact._from_view(FourInts, [0, 0, 42, 1000], list, self.provenance) fp = os.path.join(self.test_dir.name, 'artifact.qza') artifact.save(fp) with self.assertRaisesRegex( TypeError, 'Visualization does not support.*FourInts'): Visualization.extract(fp, self.test_dir)
def test_extract(self): fp = os.path.join(self.test_dir.name, 'visualization.qzv') visualization = Visualization._from_data_dir(self.data_dir) visualization.save(fp) root_dir = str(visualization.uuid) output_dir = os.path.join(self.test_dir.name, 'viz-extract-test') result_dir = Visualization.extract(fp, output_dir=output_dir) self.assertEqual(result_dir, os.path.join(output_dir, root_dir)) expected = { 'VERSION', 'metadata.yaml', 'data/index.html', 'data/css/style.css' } self.assertExtractedArchiveMembers(output_dir, root_dir, expected)
def test_extract(self): fp = os.path.join(self.test_dir.name, 'visualization.qzv') visualization = Visualization._from_data_dir(self.data_dir, self.provenance) visualization.save(fp) output_dir = os.path.join(self.test_dir.name, 'viz-extract-test') result_dir = Visualization.extract(fp, output_dir=output_dir) self.assertEqual(result_dir, output_dir) contents = [ 'visualization/VERSION', 'visualization/metadata.yaml', 'visualization/README.md', 'visualization/data/index.html', 'visualization/data/css/style.css'] for fp in contents: expected_fp = os.path.join(output_dir, fp) self.assertTrue(os.path.exists(expected_fp), 'File %s was not extracted.' % fp)
def test_extract(self): fp = os.path.join(self.test_dir.name, 'visualization.qzv') visualization = Visualization._from_data_dir( self.data_dir, self.make_provenance_capture()) visualization.save(fp) root_dir = str(visualization.uuid) output_dir = os.path.join(self.test_dir.name, 'viz-extract-test') result_dir = Visualization.extract(fp, output_dir=output_dir) self.assertEqual(result_dir, os.path.join(output_dir, root_dir)) expected = { 'VERSION', 'metadata.yaml', 'data/index.html', 'data/css/style.css', 'provenance/metadata.yaml', 'provenance/VERSION', 'provenance/action/action.yaml' } self.assertExtractedArchiveMembers(output_dir, root_dir, expected)