Python main Examples

Example #1

File: test_import_representatives.py Project: njoyard/django-representatives

def test_parltrack_import_representatives():
    fixture = os.path.join(os.path.dirname(__file__),
            'representatives_fixture.json')
    expected = os.path.join(os.path.dirname(__file__),
            'representatives_expected.json')

    # Disable django auto fields
    exclude = ('id', '_state', 'created', 'updated', 'fingerprint')

    with open(fixture, 'r') as f:
        import_representatives.main(f)

    missing = []
    with open(expected, 'r') as f:
        for obj in Deserializer(f.read()):
            compare = copy.copy(obj.object.__dict__)

            for field in exclude:
                if field in compare:
                    compare.pop(field)

            try:
                type(obj.object).objects.get(**compare)
            except:
                missing.append(compare)

    assert len(missing) is 0
    assert Representative.objects.count() == 2

Example #2

File: test_import_representatives.py Project: tomjorquera/memopol

    def test_parltrack_import_representatives(self):
        fixture = os.path.join(os.path.dirname(__file__),
                'representatives_fixture.json')
        expected = os.path.join(os.path.dirname(__file__),
                'representatives_expected.json')

        with open(fixture, 'r') as f:
            import_representatives.main(f)

        self.assertObjectsFromFixture(expected)
        assert Representative.objects.count() == 2