"""Tests for PythonExporter"""

# Copyright (c) IPython Development Team.
# Distributed under the terms of the Modified BSD License.

from .base import ExportersTestsBase
from ..python import PythonExporter


class TestPythonExporter(ExportersTestsBase):
    """Tests for PythonExporter"""

    exporter_class = PythonExporter
    should_include_raw = ['python']

    def test_constructor(self):
        """Can a PythonExporter be constructed?"""
        self.exporter_class()

    def test_export(self):
        """Can a PythonExporter export something?"""
        (output, resources) = self.exporter_class().from_filename(self._get_notebook())
        self.assertIn("coding: utf-8", output)
        self.assertIn("#!/usr/bin/env python", output)