e6839888aa4beb86b59fd79dd9ed951bc1bfb049,tests/mapper/test_link_mappers.py,Test_GraphSAGELinkGenerator,test_GraphSAGELinkGenerator_unsupervisedSampler_sample_generation,#Test_GraphSAGELinkGenerator#,457

Before Change

        unsupervisedSamples = UnsupervisedSampler(G)

        mapper = GraphSAGELinkGenerator(
            G, batch_size=self.batch_size, num_samples=self.num_samples
        ).flow(unsupervisedSamples)

        assert mapper.data_size == 16
        assert self.batch_size == 2
        assert len(mapper) == 8

After Change

        gen = GraphSAGELinkGenerator(
            G, batch_size=self.batch_size, num_samples=self.num_samples
        )
        mapper = gen.flow(unsupervisedSamples)

        assert mapper.data_size == 16
        assert self.batch_size == 2
        assert len(mapper) == 8

Instances