import unittest
from virtual2 import *

class Virtual2Test(unittest.TestCase):

    def testIt(self):
        a = A()
        self.assertEqual(a.f1(), 10)
        b = B()
        self.assertEqual(b.f1(), 10)
        self.assertEqual(b.f2(), 20)
        self.assertEqual(call_fs(b), 30)
        self.assertEqual(call_f(a), 0)
        self.assertEqual(call_f(b), 1)

        class C(B):
            def f1(self): return 1
            def f2(self): return 2
            def f(self): return 100

        c = C()
        self.assertEqual(call_fs(c), 3)
        self.assertEqual(call_fs(c), 3)
        self.assertEqual(call_f(c), 100)


if __name__ == '__main__':
    unittest.main()