summaryrefslogtreecommitdiff
path: root/lib/python2.7/site-packages/south/tests/freezer.py
blob: 82c44024af51d3d9c812f4b13c1af9e128cb94d2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from south.tests import unittest

from south.creator.freezer import model_dependencies
from south.tests.fakeapp import models

class TestFreezer(unittest.TestCase):
    def test_dependencies(self):
        self.assertEqual(set(model_dependencies(models.SubModel)),
                         set([models.BaseModel, models.Other1, models.Other2]))

        self.assertEqual(set(model_dependencies(models.CircularA)),
                         set([models.CircularA, models.CircularB, models.CircularC]))

        self.assertEqual(set(model_dependencies(models.Recursive)),
                         set([models.Recursive]))