Accept any ordering given by ListDir in the assert.

Addresses google-deepmind#168. PiperOrigin-RevId: 395907797 Change-Id: I28c7e951211c9cf6bd5e35d8dfaf57f22e843ed0
brianloyal · Sep 10, 2021 · 1d43aaf · 1d43aaf
1 parent 50ecd75
commit 1d43aaf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_alphafold_test.py b/run_alphafold_test.py
@@ -76,7 +76,7 @@ def test_end_to_end(self):
  self.assertIn('test', base_output_files)
 
  target_output_files = os.listdir(os.path.join(out_dir, 'test'))
- self.assertSequenceEqual(
+ self.assertCountEqual(
  ['features.pkl', 'msas', 'ranked_0.pdb', 'ranking_debug.json',
  'relaxed_model1.pdb', 'result_model1.pkl', 'timings.json',
  'unrelaxed_model1.pdb'], target_output_files)