import unittest

from proj import ClassA
from proj.moduleB import ClassB

class Inclusion(unittest.TestCase):

    # Both classes are accessable
    def test_A(self):
        m = ClassA('x')
        self.assertEqual('x', m.get_x())

    def test_B(self):
        m = ClassB('x')
        self.assertEqual('x', m.get_x())

    # Can't import ClassB directly, since it's not mentioned in proj/__init__.py
    def test_direct_class(self):
        with self.assertRaises(ImportError):
            from proj import ClassB

8ad751f1ffadcac2a030e71c37776a57e81920a3
<mask token>


class Inclusion(unittest.TestCase):

    def test_A(self):
        m = ClassA('x')
        self.assertEqual('x', m.get_x())
    <mask token>
    <mask token>

<mask token>


class Inclusion(unittest.TestCase):

    def test_A(self):
        m = ClassA('x')
        self.assertEqual('x', m.get_x())
    <mask token>

    def test_direct_class(self):
        with self.assertRaises(ImportError):
            from proj import ClassB

<mask token>


class Inclusion(unittest.TestCase):

    def test_A(self):
        m = ClassA('x')
        self.assertEqual('x', m.get_x())

    def test_B(self):
        m = ClassB('x')
        self.assertEqual('x', m.get_x())

    def test_direct_class(self):
        with self.assertRaises(ImportError):
            from proj import ClassB

import unittest
from proj import ClassA
from proj.moduleB import ClassB


class Inclusion(unittest.TestCase):

    def test_A(self):
        m = ClassA('x')
        self.assertEqual('x', m.get_x())

    def test_B(self):
        m = ClassB('x')
        self.assertEqual('x', m.get_x())

    def test_direct_class(self):
        with self.assertRaises(ImportError):
            from proj import ClassB

import unittest

from proj import ClassA
from proj.moduleB import ClassB

class Inclusion(unittest.TestCase):

    # Both classes are accessable
    def test_A(self):
        m = ClassA('x')
        self.assertEqual('x', m.get_x())

    def test_B(self):
        m = ClassB('x')
        self.assertEqual('x', m.get_x())

    # Can't import ClassB directly, since it's not mentioned in proj/__init__.py
    def test_direct_class(self):
        with self.assertRaises(ImportError):
            from proj import ClassB
