示例#1
0
    def test_several_inspections_several_parcels_same_tract(self):
        crimes = [("16Sep2014", "tract567", 3),
                  ("18Oct2014", "tract567", 1)]
        parcels = [("parcelA", "01Dec2014", "tract567"),
                   ("parcelB", "18Nov2014", "tract567"),]
        population = [("tract567", 1234)]
        window = datetime.timedelta(days=365)

        expected = [("parcelA", timestamp("01Dec2014"), 4 / float(1234)),
                    ("parcelB", timestamp("18Nov2014"), 4 / float(1234))]

        actual = crime_agg.crimerate_in_aggregation_area(self.make_parcels_df(parcels), self.make_crimerate_df(crimes),
                                       self.make_population_df(population), window)
        actual = actual.reset_index()[["parcel_id", "inspection_date", "crime_rate"]].values

        self.assert_array_almost_equal(expected, actual)
    def test_several_inspections_same_parcel_one_crime_in_range(self):
        crimes = [("16Sep2014", "tract567", 3), ("18Dec2014", "tract567", 1)]
        parcels = [("parcelA", "01Dec2014", "tract567"),
                   ("parcelA", "18Nov2014", "tract567")]
        population = [("tract567", 1234)]
        window = datetime.timedelta(days=365)

        expected = [("parcelA", timestamp("01Dec2014"), 3 / float(1234)),
                    ("parcelA", timestamp("18Nov2014"), 3 / float(1234))]

        actual = crime_agg.crimerate_in_aggregation_area(
            self.make_parcels_df(parcels), self.make_crimerate_df(crimes),
            self.make_population_df(population), window)
        actual = actual.reset_index()[[
            "parcel_id", "inspection_date", "crime_rate"
        ]].values

        self.assert_array_almost_equal(expected, actual)
示例#3
0
    def test_six_month_window(self):
        crimes = [("16Sep2014", "tract567", 3),
                  ("18Sep2014", "tract568", 1),
                  ("14Mar2014", "tract568", 6)]
        parcels = [("parcelA", "01Dec2014", "tract567"),
                   ("parcelB", "18Apr2014", "tract568"),]
        population = [("tract567", 1234),
                      ("tract568", 234)]
        window = datetime.timedelta(days=180)

        expected = [("parcelA", timestamp("01Dec2014"), 3 / float(1234)),
                    ("parcelB", timestamp("18Apr2014"), 6 / float(234))]

        actual = crime_agg.crimerate_in_aggregation_area(self.make_parcels_df(parcels), self.make_crimerate_df(crimes),
                                       self.make_population_df(population), window)
        actual = actual.reset_index()[["parcel_id", "inspection_date", "crime_rate"]].values

        self.assert_array_almost_equal(expected, actual)
        assert_array_equal(expected, actual)
    def test_six_month_window(self):
        crimes = [("16Sep2014", "tract567", 3), ("18Sep2014", "tract568", 1),
                  ("14Mar2014", "tract568", 6)]
        parcels = [
            ("parcelA", "01Dec2014", "tract567"),
            ("parcelB", "18Apr2014", "tract568"),
        ]
        population = [("tract567", 1234), ("tract568", 234)]
        window = datetime.timedelta(days=180)

        expected = [("parcelA", timestamp("01Dec2014"), 3 / float(1234)),
                    ("parcelB", timestamp("18Apr2014"), 6 / float(234))]

        actual = crime_agg.crimerate_in_aggregation_area(
            self.make_parcels_df(parcels), self.make_crimerate_df(crimes),
            self.make_population_df(population), window)
        actual = actual.reset_index()[[
            "parcel_id", "inspection_date", "crime_rate"
        ]].values

        self.assert_array_almost_equal(expected, actual)
        assert_array_equal(expected, actual)
示例#5
0
    def test_one_inspection_empty_crimes(self):
        crimes = []
        parcels = [("parcelA", "01Dec2014", "tract567")]
        population = [("tract567", 1234)]
        window = datetime.timedelta(days=365)

        expected = [("parcelA", timestamp("01Dec2014"), 0.0)]

        actual = crime_agg.crimerate_in_aggregation_area(self.make_parcels_df(parcels), self.make_crimerate_df(crimes),
                                       self.make_population_df(population), window)
        actual = actual.reset_index()[["parcel_id", "inspection_date", "crime_rate"]].values

        self.assert_array_almost_equal(expected, actual)
    def test_one_inspection_empty_crimes(self):
        crimes = []
        parcels = [("parcelA", "01Dec2014", "tract567")]
        population = [("tract567", 1234)]
        window = datetime.timedelta(days=365)

        expected = [("parcelA", timestamp("01Dec2014"), 0.0)]

        actual = crime_agg.crimerate_in_aggregation_area(
            self.make_parcels_df(parcels), self.make_crimerate_df(crimes),
            self.make_population_df(population), window)
        actual = actual.reset_index()[[
            "parcel_id", "inspection_date", "crime_rate"
        ]].values

        self.assert_array_almost_equal(expected, actual)