85f717770691b70cfa9077d2d3228b205d19b18d,tests/test_preprocessing.py,TestPreprocessing,test_rotate_pyfunc,#TestPreprocessing#,152

Before Change

            self.x_and_pi_same(run_one, run_two),
            "Not randomly rotated")

        syms = []
        for (x, pi, v), (x2, pi2, v2) in zip(run_one, run_two):
            self.assertEqual(v, v2, "values not the same")
            // For each record find the symmetry that makes them equal
            syms.extend(
                map(lambda r: self.find_symmetry(*r), zip(x, pi, x2, pi2)))

        difference = set(symmetries.SYMMETRIES) - set(syms)
        self.assertEqual(len(syms), num_records, "Not same number of records")
        self.assertEqual(difference, set(), "Didn"t find these rotations")

After Change

            self.reset_random()
            run_three = self.extract_data(f.name, random_rotation=True)

        self.assert_rotate_data(run_one, run_two, run_three)

    def test_tpu_rotate(self):
        num_records = 100
        raw_data = self.create_random_data(num_records)

Instances