def test_handle_errors_dict(self, hf3, hf4, hf5, errors):
errors.in_2010 = {'11001000100': '22002000200', '11001000902': None}
# Create Mock GEO file
shutil.copyfile(os.path.join("censusdata", "tests", "mock_geo.txt"),
os.path.join(self.tempdir, "ZZgeo2010.sf1"))
command = Command()
command.handle(os.path.join(self.tempdir, 'ZZgeo2010.sf1'))
positional_args = hf4.call_args[0]
# The None causes us to skip 11001000902
self.assertEqual(len(positional_args[2]), 1)
# This entry was converted
self.assertEqual(positional_args[2]['0007159'], '22002000200')
def test_handle(self, hft, hff):
# Create Mock GEO file
shutil.copyfile(os.path.join("censusdata", "tests", "mock_geo.txt"),
os.path.join(self.tempdir, "ZZgeo2010.sf1"))
command = Command()
command.handle(os.path.join(self.tempdir, 'ZZgeo2010.sf1'))
positional_args = hft.call_args[0]
self.assertEqual(positional_args[0],
os.path.join(self.tempdir, "ZZgeo2010.sf1"))
self.assertEqual(len(positional_args[1]), 2)
self.assertEqual(positional_args[1]['0007159'], '11001000100')
self.assertEqual(positional_args[1]['0007211'], '11001000902')
def test_handle(self, hf3, hf4, hf5):
# Create Mock GEO file
shutil.copyfile(os.path.join("censusdata", "tests", "mock_geo.txt"),
os.path.join(self.tempdir, "ZZgeo2010.sf1"))
command = Command()
command.handle(os.path.join(self.tempdir, 'ZZgeo2010.sf1'))
positional_args = hf4.call_args[0]
self.assertEqual(positional_args[0],
os.path.join(self.tempdir, "ZZgeo2010.sf1"))
self.assertEqual(positional_args[1], '11') # State
self.assertEqual(len(positional_args[2]), 2)
self.assertEqual(positional_args[2]['0007159'], '11001000100')
self.assertEqual(positional_args[2]['0007211'], '11001000902')
def test_handle(self, hf3, hf4, hf5):
# Create Mock GEO file
year = '2013'
shutil.copyfile(os.path.join("censusdata", "tests", "mock_geo.txt"),
os.path.join(self.tempdir, "ZZgeo2010.sf1"))
command = Command()
command.handle(os.path.join(self.tempdir, 'ZZgeo2010.sf1'), year)
positional_args = hf4.call_args[0]
self.assertEqual(positional_args[0],
os.path.join(self.tempdir, "ZZgeo2010.sf1"))
self.assertEqual(positional_args[1], year)
self.assertEqual(positional_args[2], '11') # State
self.assertEqual(len(positional_args[3]), 2)
self.assertEqual(positional_args[3]['0007159'], year+'11001000100')
self.assertEqual(positional_args[3]['0007211'], year+'11001000902')
