def test_arithmetic(): a = Varying(base_time=2, slope=3, base=5) b = Varying(base_time=7, slope=11, base=13) add = a + b assert add._base_time == a._base_time assert a(20) + b(20) == add(20) assert a(30) + b(30) == add(30) sub = a - b assert add._base_time == a._base_time assert a(20) - b(20) == sub(20) assert a(30) - b(30) == sub(30) neg = -a assert neg._base_time == a._base_time assert a(20) == -neg(20) assert a(30) == -neg(30) offset1 = a + 2 offset2 = 2 + a assert add._base_time == a._base_time assert a(20) + 2 == offset1(20) == offset2(20) assert a(30) + 2 == offset1(30) == offset2(30) dif1 = a - 2 dif2 = 2 - a assert add._base_time == a._base_time assert a(20) - 2 == dif1(20) == -dif2(20) assert a(30) - 2 == dif1(30) == -dif2(30) double1 = a * 2 double2 = 2 * a assert double1._base_time == a._base_time assert double2._base_time == a._base_time assert double1(20) == double2(20) == a(20) * 2 assert double1(30) == double2(30) == a(30) * 2 half = a / 2 assert half._base_time == a._base_time assert half(20) == a(20) / 2 assert half(30) == a(30) / 2 with pytest.raises(TypeError): _ = Varying.T + 'test' with pytest.raises(TypeError): _ = 'test' + Varying.T with pytest.raises(TypeError): _ = 'test' - Varying.T with pytest.raises(TypeError): _ = Varying.T - 'test' with pytest.raises(TypeError): _ = Varying.T * 'test' with pytest.raises(TypeError): _ = 'test' * Varying.T with pytest.raises(TypeError): _ = Varying.T / 'test'
def create_blossom(self, contained_region_ids: List[int]) -> int: k = self._next_region_id self._next_region_id += 1 blossom_region = GraphFillRegion(id=k) blossom_region.radius = Varying(base_time=self.time, slope=1) blossom_region.blossom_children = [] self._region_data_map[k] = blossom_region for i in contained_region_ids: region = self._region_data_map[i] del self._region_data_map[i] blossom_region.blossom_children.append(region) region.radius = region.radius.then_slope_at( time_of_change=self.time, new_slope=0) for loc in region.area: blossom_region.area.add(loc) loc_data = self._loc_data(loc) loc_data.mile_markers.append( MileMarker( blossom_region, -loc_data.local_radius()(self.time), )) self._reschedule_events_for_region(blossom_region) # Rescheduling the blossom region fixed location schedules, but not # child region schedules. Fix them now. for child in blossom_region.blossom_children: for ev in list(child.schedule_map.values()): ev.invalidate() return k
def test_zero_intercept(): assert Varying.T.zero_intercept() == 0 assert (Varying.T + 5).zero_intercept() == -5 assert (-Varying.T * 2 + 5).zero_intercept() == 2.5 assert (-Varying.T * 0 + 5).zero_intercept() is None assert Varying(base_time=5, base=3, slope=2).zero_intercept() == 3.5 assert (Varying.T * 0).zero_intercept() is None
def test_scalar(): with pytest.raises(TypeError): _ = int(Varying(slope=1)) with pytest.raises(TypeError): _ = float(Varying(slope=1)) with pytest.raises(TypeError): _ = complex(Varying(slope=1)) assert float(Varying(2)) == 2 assert int(Varying(2)) == 2 assert complex(Varying(2)) == 2 assert isinstance(float(Varying(2)), float) assert isinstance(int(Varying(2)), int) assert isinstance(complex(Varying(2)), complex)
def create_region(self, location: complex) -> int: k = self._next_region_id self._next_region_id += 1 self.regions[k] = CircleFillRegion(source=complex(location), id=k, radius=Varying(base_time=self.time, slope=1)) return k
def __init__( self, *, id: int, radius: Union[int, float, Varying], source: Union[None, int, float, complex] = None, blossom_children: Optional[List['CircleFillRegion']] = None, ): self.id = id self.source = None if source is None else complex(source) self.radius = Varying(radius) self.blossom_children = blossom_children
def test_init(): z = Varying() assert z._base_time == 0 assert z._base == 0 assert z.slope == 0 assert z(0) == 0 assert z(1) == 0 z = Varying(base=2, base_time=3, slope=5) assert z._base_time == 3 assert z._base == 2 assert z.slope == 5 assert z(0) == -13 assert z(1) == -8 z = Varying(z) assert z._base_time == 3 assert z._base == 2 assert z.slope == 5 assert z(0) == -13 assert z(1) == -8
def __init__( self, *, id: int, source: Optional[TLocation] = None, radius: Union[int, float, Varying] = Varying.T, blossom_children: Optional[Iterable['GraphFillRegion']] = None, ): self.id = id self.source = source self.area: Set[TLocation] = set() self.radius = Varying(radius) self.blossom_children: Optional[List['GraphFillRegion', ...]] = ( None if blossom_children is None else list(blossom_children)) self.schedule_map: Dict[int, TentativeEvent] = {}
def create_blossom(self, contained_region_ids: List[int]) -> int: regions = [] for i in contained_region_ids: regions.append(self.regions[i]) del self.regions[i] for r in regions: r.radius = r.radius.then_slope_at(time_of_change=self.time, new_slope=0) k = self._next_region_id self._next_region_id += 1 self.regions[k] = CircleFillRegion(id=k, blossom_children=regions, radius=Varying(base_time=self.time, slope=1)) return k
def test_normal_progression(): def assert_process_event(*expected_commands: Any): event = fill.next_event() mwpm.process_event(event) assert fill.recorded_commands == list(expected_commands) fill.recorded_commands.clear() fill = RecordingFlooder(CircleFlooder()) mwpm = Mwpm(flooder=fill) mwpm.add_region(fill.create_region(100)) mwpm.add_region(fill.create_region(101)) mwpm.add_region(fill.create_region(200)) mwpm.add_region(fill.create_region(202)) mwpm.add_region(fill.create_region(300)) fill.recorded_commands.clear() # Pair ups. assert_process_event( ('next_event', RegionHitRegionEvent(region1=0, region2=1, time=0.5)), ('set_region_growth', 0, 0), ('set_region_growth', 1, 0), ) assert_process_event( ('next_event', RegionHitRegionEvent(region1=2, region2=3, time=1)), ('set_region_growth', 2, 0), ('set_region_growth', 3, 0), ) # Alternating tree starts. assert_process_event( ('next_event', RegionHitRegionEvent(region1=3, region2=4, time=97)), ('set_region_growth', 3, -1), ('set_region_growth', 2, +1), ) # Alternating tree turns into a blossom. assert_process_event( ('next_event', RegionHitRegionEvent(region1=2, region2=4, time=98)), ('create_blossom', (2, 3, 4), 5), ) assert fill.sub_flooder.regions == { 0: CircleFillRegion(id=0, source=100, radius=0.5), 1: CircleFillRegion(id=1, source=101, radius=0.5), 5: CircleFillRegion( id=5, radius=Varying(base_time=98, slope=1), blossom_children=[ CircleFillRegion(id=2, source=200, radius=2), CircleFillRegion(id=3, source=202, radius=0), CircleFillRegion(id=4, source=300, radius=98), ], ), } assert_process_event( ('next_event', RegionHitRegionEvent(region1=1, region2=5, time=194.5)), ('set_region_growth', 1, -1), ('set_region_growth', 0, +1), ) assert_process_event( ('next_event', RegionHitRegionEvent(region1=0, region2=5, time=195)), ('create_blossom', (0, 1, 5), 6), ) assert fill.next_event() is None
def test_blossom_implosion(): def assert_process_event(*expected_commands: Any): event = fill.next_event() mwpm.process_event(event) assert fill.recorded_commands == list(expected_commands) fill.recorded_commands.clear() fill = RecordingFlooder(CircleFlooder()) mwpm = Mwpm(flooder=fill) mwpm.add_region(fill.create_region(0)) mwpm.add_region(fill.create_region(1)) mwpm.add_region(fill.create_region(3)) mwpm.add_region(fill.create_region(-10)) mwpm.add_region(fill.create_region(+10)) fill.recorded_commands.clear() # Blossom created in center. assert_process_event( ('next_event', RegionHitRegionEvent(region1=0, region2=1, time=0.5)), ('set_region_growth', 0, 0), ('set_region_growth', 1, 0), ) assert_process_event( ('next_event', RegionHitRegionEvent(region1=1, region2=2, time=1.5)), ('set_region_growth', 1, -1), ('set_region_growth', 0, +1), ) assert_process_event( ('next_event', RegionHitRegionEvent(region1=0, region2=2, time=2)), ('create_blossom', (0, 1, 2), 5), ) assert fill.sub_flooder.regions == { 3: CircleFillRegion(id=3, source=-10, radius=Varying.T), 4: CircleFillRegion(id=4, source=+10, radius=Varying.T), 5: CircleFillRegion( id=5, radius=Varying(base_time=2, slope=1), blossom_children=[ CircleFillRegion(id=0, source=0, radius=1), CircleFillRegion(id=1, source=1, radius=0), CircleFillRegion(id=2, source=3, radius=2), ], ), } # Blossom becomes an inner node. assert_process_event( ('next_event', RegionHitRegionEvent(region1=4, region2=5, time=3.5)), ('set_region_growth', 4, 0), ('set_region_growth', 5, 0), ) assert_process_event( ('next_event', RegionHitRegionEvent(region1=3, region2=5, time=7.5)), ('set_region_growth', 5, -1), ('set_region_growth', 4, +1), ) # Blossom implodes. assert_process_event( ( 'next_event', BlossomImplodeEvent(blossom_region_id=5, time=9, in_out_touch_pairs=[ (0, 3), (2, 4), ]), ), ('set_region_growth', 0, -1), ('set_region_growth', 1, +1), ('set_region_growth', 2, -1), )
def local_radius(self) -> Varying: if self.mile_markers: marker = self.mile_markers[-1] return marker.region.radius - marker.distance_from_region_center return Varying(0)
def test_approximate_equality(): assert cirq.approx_eq(Varying(5), Varying(5.01), atol=0.1) assert not cirq.approx_eq(Varying(5), Varying(5.01), atol=0.001) assert cirq.approx_eq(Varying(5, slope=4), Varying(5, slope=4.01), atol=0.1) assert not cirq.approx_eq(Varying(5, slope=4), Varying(5, slope=4.01), atol=0.001) assert not cirq.approx_eq(Varying.T, 'test', atol=0.1) assert cirq.approx_eq(Varying(5), 5.001, atol=0.01) assert cirq.approx_eq(5.001, Varying(5), atol=0.01) assert not cirq.approx_eq(Varying(5), 5.001, atol=0.0001) assert not cirq.approx_eq(5.001, Varying(5), atol=0.0001)
def test_equality(): eq = cirq.testing.EqualsTester() eq.add_equality_group( Varying(base_time=0, slope=1, base=0), Varying(base_time=1, slope=1, base=1), Varying(Varying(base_time=0, slope=1, base=0)), ) eq.add_equality_group( Varying(base_time=0, slope=-1, base=0), Varying(base_time=1, slope=-1, base=-1), -Varying.T, ) eq.add_equality_group( Varying(base_time=0, slope=1, base=1), Varying(base_time=1, slope=1, base=2), ) eq.add_equality_group( Varying(base_time=0, slope=0, base=50), Varying(base_time=10000, slope=0, base=50), 50, 50.0, 50.0 + 0j, ) eq.add_equality_group( Varying(base_time=0, slope=0, base=60), Varying(base_time=10000, slope=0, base=60), 60, )
def test_repr(): cirq.testing.assert_equivalent_repr( Varying(base_time=2, slope=3, base=5), global_vals={'Varying': Varying} )
def test_then_slope_at(): a = Varying(base_time=2, slope=3, base=5) b = a.then_slope_at(time_of_change=7, new_slope=11) assert b._base_time == 7 assert b(7) == a(7) assert b.slope == 11