Added state, county, and tract fields to the Block class.

[dead/census-tools.git] / src / Tests / Unit / SummaryFile1Test.py

diff --git a/src/Tests/Unit/SummaryFile1Test.py b/src/Tests/Unit/SummaryFile1Test.py
index a29e208cf7fa91ee2bbc20ce186b9f01e1d63d02..edaf59859d91e1e89a5026ee959d1b4b38a87c6a 100644 (file)
--- a/src/Tests/Unit/SummaryFile1Test.py
+++ b/src/Tests/Unit/SummaryFile1Test.py
@@ -12,6 +12,9 @@ class BlockTest(unittest.TestCase):
         # Fill a GeoRecord with dummy values so that we
         # can calculate its average population density.
         gr = SummaryFile1.GeoRecord()
+        gr.state = '24'
+        gr.county = '001'
+        gr.tract = '123456'
         gr.block = '1728'
         gr.pop100 = '100'
         gr.arealand = '40'
@@ -80,7 +83,7 @@ class SummaryFile1Test(unittest.TestCase):
             b_coords.longitude = b.coordinates.longitude
             
             closest_block = SummaryFile1.FindClosestBlock(blocks, b_coords)
-            self.assertEqual(b.block_number, closest_block.block_number)
+            self.assertEqual(b.block, closest_block.block)
 
 
     def testEachBlockHasItsOwnAverageDensity(self):