예제 #1
0
    def test_multi_files(self):

        command = Command()
        command.stdout = Mock()

        main_csv_directory = os.path.abspath(os.path.join("hmda", "tests"))

        main_csv_directory = main_csv_directory + "/"

        command.handle(main_csv_directory, "delete_file:false", "filterhmda")

        lenders = set(r.institution_id for r in HMDARecord.objects.all())
        geos = set(r.geo_id for r in HMDARecord.objects.all())

        self.assertEqual(3, len(lenders))
        self.assertTrue(('5' + '0000000319') in lenders)
        self.assertTrue(('5' + '0000000435') in lenders)
        self.assertTrue(('3' + '0000001281') in lenders)
        self.assertEqual(4, len(geos))
        self.assertTrue('1122233300' in geos)
        self.assertTrue('1122233400' in geos)
        self.assertTrue('1122333300' in geos)
        self.assertTrue('1222233300' in geos)

        HMDARecord.objects.all().delete()
예제 #2
0
    def test_multi_files(self):

        command = Command()
        command.stdout = Mock()

        main_csv_directory = os.path.abspath( os.path.join("hmda", "tests") )

        main_csv_directory = main_csv_directory + "/"

        command.handle(main_csv_directory , "delete_file:false", "filterhmda" )

        lenders = set(r.lender for r in HMDARecord.objects.all())
        geos = set(r.geoid_id for r in HMDARecord.objects.all())

        self.assertEqual(3, len(lenders))
        self.assertTrue(('5' + '0000000319') in lenders)
        self.assertTrue(('5' + '0000000435') in lenders)
        self.assertTrue(('3' + '0000001281') in lenders)
        self.assertEqual(4, len(geos))
        self.assertTrue('1122233300' in geos)
        self.assertTrue('1122233400' in geos)
        self.assertTrue('1122333300' in geos)
        self.assertTrue('1222233300' in geos)

        HMDARecord.objects.all().delete()
예제 #3
0
    def test_handle_errors_dict(self, errors):
        errors.in_2010 = {'1122233300': '9988877766'}
        command = Command()
        command.stdout = Mock()
        command.handle(os.path.join("hmda", "tests", "mock_2014.csv"))

        geos = set(r.geoid_id for r in HMDARecord.objects.all())
        self.assertEqual(4, len(geos))
        # 1122233300 got replaced
        self.assertTrue('9988877766' in geos)
        self.assertFalse('1122233300' in geos)

        HMDARecord.objects.all().delete()
예제 #4
0
    def test_handle(self):
        command = Command()
        command.stdout = Mock()
        command.handle(os.path.join("hmda", "tests", "mock_2014.csv"))

        # The mock data file contains 10 records, 8 for known states
        self.assertEqual(8, HMDARecord.objects.count())
        lenders = set(r.lender for r in HMDARecord.objects.all())
        geos = set(r.geoid_id for r in HMDARecord.objects.all())
        self.assertEqual(3, len(lenders))
        self.assertTrue(('5' + '0000000319') in lenders)
        self.assertTrue(('5' + '0000000435') in lenders)
        self.assertTrue(('3' + '0000001281') in lenders)
        self.assertEqual(4, len(geos))
        self.assertTrue('1122233300' in geos)
        self.assertTrue('1122233400' in geos)
        self.assertTrue('1122333300' in geos)
        self.assertTrue('1222233300' in geos)

        HMDARecord.objects.all().delete()