Пример #1
0
    def test_handle_filefive(self):
        shutil.copyfile(os.path.join("censusdata", "tests", "mock_file5.txt"),
                        os.path.join(self.tempdir, "ZZ000052010.sf1"))
        command = Command()
        command.handle_filefive(
            os.path.join(self.tempdir, "ZZgeo2010.sf1"),
            '2013',
            '11',  # State
            {
                '0007159': '11001000100',
                '0007211': '11001000902'
            })
        model = models.Census2010Households.objects.get(pk='11001000100')
        self.assertEqual(model.total, 1853624)
        self.assertEqual(model.total_family, 1067203)
        self.assertEqual(model.total_nonfamily, 786421)
        model.delete()

        model = models.Census2010Households.objects.get(pk='11001000902')
        self.assertEqual(model.total, 2738936)
        self.assertEqual(model.total_family, 1951351)
        self.assertEqual(model.total_nonfamily, 787585)
        model.delete()

        self.assertEqual(len(models.Census2010Households.objects.all()), 0)
Пример #2
0
    def test_handle_filefive(self):
        shutil.copyfile(os.path.join("censusdata", "tests", "mock_file5.txt"),
                        os.path.join(self.tempdir, "ZZ000052010.sf1"))
        command = Command()
        command.handle_filefive(os.path.join(self.tempdir, "ZZgeo2010.sf1"),
                                '11',  # State
                                {'0007159': '11001000100',
                                 '0007211': '11001000902'})
        model = models.Census2010Households.objects.get(pk='11001000100')
        self.assertEqual(model.total, 1853624)
        self.assertEqual(model.total_family, 1067203)
        self.assertEqual(model.total_nonfamily, 786421)
        model.delete()

        model = models.Census2010Households.objects.get(pk='11001000902')
        self.assertEqual(model.total, 2738936)
        self.assertEqual(model.total_family, 1951351)
        self.assertEqual(model.total_nonfamily, 787585)
        model.delete()

        self.assertEqual(len(models.Census2010Households.objects.all()), 0)