def test_no_redirect_html_response(self, mock_make_grammar_info): dset = Dataset(name='blank', user='******') dset.global_stats_calculated = True dset.classical = False dset.grammar_navbar['lengths'] = [1, 3, 6] dset.global_stats = { 'grams': "[frozenset([(1, 2)]), frozenset([2, 3])]", 'num_poots': 5, 'num_total_poots': 19, 'percent_poots': 5.0/19, 'num_cots': 2, 'num_total_cots': 6, 'percent_cots': 2.0/6 } dset.save() response = self.client.get(url_for( 'grammars.global_stats_calculated', dset_name='blank', sort_value=6, classical=False, page=0, sort_by='size' )) self.assert_200(response) print response.data data = json.loads(response.data) assert not data['need_redirect'] assert data['finished'] assert not data['retry'] assert data['html_str'] assert data['grammar_stat_url'] mock_make_grammar_info.assert_called_with('blank')
def test_no_grammars_exist(self, mock_make_grammar_info): dset = Dataset(name='blank', user='******') dset.global_stats_calculated = True dset.classical = False dset.grammar_navbar['lengths'] = [] dset.global_stats = { 'grams': "[]", 'num_poots': 0, 'num_total_poots': 19, 'percent_poots': 0.0, 'num_cots': 0, 'num_total_cots': 6, 'percent_cots': 0.0 } dset.save() response = self.client.get(url_for( 'grammars.global_stats_calculated', dset_name='blank', sort_value=0, classical=False, page=0, sort_by='size' )) print(response.data) self.assert_200(response) data = json.loads(response.data) assert not data['need_redirect'] assert data['finished'] assert not data['retry'] assert not data['grammars_exist'] assert not mock_make_grammar_info.called